Update app.py

app.py

@@ -343,6 +343,26 @@ class ImageStoryteller:
         scene_type = analysis['scenes'][0]['type'] if analysis['scenes'] else "unknown"
         
         return story, detected_objects, scene_type
+
+    def create_story_overlay(self, image, story):
+    """Create formatted text with caption and story for textbox display"""
+    
+    # Generate caption (first sentence of the story)
+    caption = ""
+    sentences = story.split('. ')
+    if sentences:
+        caption = sentences[0].strip()
+        if not caption.endswith('.'):
+            caption += '.'
+    
+    # Format the text with caption separated from story
+    # Using a separator line of dashes
+    separator = "─" * 40
+    
+    # Format the complete text for the textbox
+    formatted_text = f"{caption}\n{separator}\n{story}"
+    
+    return formatted_text
     
     # FIXED: Added the missing method
     # def create_story_overlay(self, image, story):
@@ -351,58 +371,58 @@ class ImageStoryteller:
     #     # For now, let's just return the story text
     #     return story
 
-    def create_story_overlay(self, image, story):
-        """Create story overlay as separate black image with bigger fonts"""
-        img_np = np.array(image)
-        height, width = 800, 800#img_np.shape[:2]
+    # def create_story_overlay(self, image, story):
+    #     """Create story overlay as separate black image with bigger fonts"""
+    #     img_np = np.array(image)
+    #     height, width = 800, 800#img_np.shape[:2]
         
-        # Create a separate black image for the story (1/3 of original height)
-        overlay_height = height // 1
-        overlay = np.zeros((overlay_height, width, 3), dtype=np.uint8)
+    #     # Create a separate black image for the story (1/3 of original height)
+    #     overlay_height = height // 1
+    #     overlay = np.zeros((overlay_height, width, 3), dtype=np.uint8)
         
-        # Add text to the black overlay with bigger fonts
-        font = cv2.FONT_HERSHEY_SIMPLEX
-        font_scale = 1  # Much bigger font
-        font_color = (255, 255, 255)  # White text
-        thickness = 1  # Thicker text
-        line_spacing = 25  # More spacing for bigger text
+    #     # Add text to the black overlay with bigger fonts
+    #     font = cv2.FONT_HERSHEY_SIMPLEX
+    #     font_scale = 1  # Much bigger font
+    #     font_color = (255, 255, 255)  # White text
+    #     thickness = 1  # Thicker text
+    #     line_spacing = 25  # More spacing for bigger text
         
-        # Split story into lines (max 40 characters per line for bigger text)
-        words = story.split()
-        lines = []
-        current_line = ""
+    #     # Split story into lines (max 40 characters per line for bigger text)
+    #     words = story.split()
+    #     lines = []
+    #     current_line = ""
         
-        for word in words:
-            if len(current_line + word) <= 40:
-                current_line += word + " "
-            else:
-                lines.append(current_line.strip())
-                current_line = word + " "
-        if current_line:
-            lines.append(current_line.strip())
+    #     for word in words:
+    #         if len(current_line + word) <= 40:
+    #             current_line += word + " "
+    #         else:
+    #             lines.append(current_line.strip())
+    #             current_line = word + " "
+    #     if current_line:
+    #         lines.append(current_line.strip())
         
-        # Limit to 5 lines maximum for bigger text
-        # if len(lines) > 5:  # If you want to keep some limit but indicate truncation
-        #     lines = lines[:5]
-        #     lines.append("... [Story continues]")  # Indicate truncation
+    #     # Limit to 5 lines maximum for bigger text
+    #     # if len(lines) > 5:  # If you want to keep some limit but indicate truncation
+    #     #     lines = lines[:5]
+    #     #     lines.append("... [Story continues]")  # Indicate truncation
         
         
-        # Calculate text block height for centering
-        total_text_height = len(lines) * line_spacing
-        start_y = (overlay_height - total_text_height) // + 60
+    #     # Calculate text block height for centering
+    #     total_text_height = len(lines) * line_spacing
+    #     start_y = (overlay_height - total_text_height) // + 60
         
-        # Add each line of text, centered
-        y_offset = start_y
-        for line in lines:
-            # Calculate text size for centering
-            text_size = cv2.getTextSize(line, font, font_scale, thickness)[0]
-            text_x = (width - text_size[0]) // 2
-            
-            cv2.putText(overlay, line, (text_x, y_offset), 
-                       font, font_scale, font_color, thickness, cv2.LINE_AA)
-            y_offset += line_spacing
+    #     # Add each line of text, centered
+    #     y_offset = start_y
+    #     for line in lines:
+    #         # Calculate text size for centering
+    #         text_size = cv2.getTextSize(line, font, font_scale, thickness)[0]
+    #         text_x = (width - text_size[0]) // 2
+            
+    #         cv2.putText(overlay, line, (text_x, y_offset), 
+    #                    font, font_scale, font_color, thickness, cv2.LINE_AA)
+    #         y_offset += line_spacing
         
-        return Image.fromarray(overlay)
+    #     return Image.fromarray(overlay)
     
     def remove_background(self, image):
         """Remove background using rembg"""
@@ -705,7 +725,7 @@ with gr.Blocks(title="Who says AI isn’t creative? Watch it turn a single image
         with gr.Column():
             story_output = gr.Textbox(
                 label="🔍 Story",
-                height=None,
+                # height=None,
                 # show_download_button=True
 
                 lines=10,