Update app.py

app.py

@@ -66,30 +66,25 @@ def text_to_speech(story):
     return audio_file_path
 
 demo = gr.Interface(
-    fn={
-        "Detect Objects": detect_objects,
-        "Generate Story": generate_story,
-        "Summarize Story": summarize_story,
-        "Generate Images": generate_images,
-        "Text to Speech": text_to_speech,
-    },
-    inputs={
-        "Detect Objects": gr.Image(type="numpy"),
-        "Generate Story": gr.Textbox(placeholder="Write a story prompt..."),
-        "Summarize Story": gr.Textbox(placeholder="Enter the generated story..."),
-        "Generate Images": gr.Textbox(placeholder="Enter the story for image generation..."),
-        "Text to Speech": gr.Textbox(placeholder="Enter text to convert to speech...")
-    },
-    outputs={
-        "Detect Objects": "text",
-        "Generate Story": "text",
-        "Summarize Story": "text",
-        "Generate Images": "image",
-        "Text to Speech": "audio"
-    },
+    fn=[detect_objects, generate_story, summarize_story, generate_images, text_to_speech],
+    inputs=[
+        gr.Image(type="numpy"),  # Detect Objects
+        gr.Textbox(placeholder="Write a story prompt..."),  # Generate Story
+        gr.Textbox(placeholder="Enter the generated story..."),  # Summarize Story
+        gr.Textbox(placeholder="Enter the story for image generation..."),  # Generate Images
+        gr.Textbox(placeholder="Enter text to convert to speech...")  # Text to Speech
+    ],
+    outputs=[
+        gr.Textbox(label="Detected Objects"),
+        gr.Textbox(label="Generated Story"),
+        gr.Textbox(label="Story Summary"),
+        gr.Image(label="Generated Images"),
+        gr.Audio(label="Generated Speech")
+    ],
     title="AI-Powered Storytelling Assistant",
     description="An AI assistant combining Object Detection, NLP, Image Generation, and TTS!"
 )
+)
 
 if __name__ == "__main__":
     demo.launch()