Update app.py

app.py

@@ -1,59 +1,61 @@
 import gradio as gr
 from gradio_client import Client
 
-# Connect to your existing Gradio apps
+# Connect to your deployed Gradio apps
 grounded_client = Client("admin08077/veo")
 image_client = Client("admin08077/picgenai")
 chat_client = Client("admin08077/genai")
 
 def generate_book(topic):
-    # Step 1: Get grounded fact-based response
+    # Step 1: Get grounded info
     try:
         grounded_response = grounded_client.predict(
             prompt=topic,
             api_name="/generate_grounded_response"
         )
     except Exception as e:
-        grounded_response = f"Error getting grounded response: {e}"
+        grounded_response = f"❌ Error getting grounded response: {e}"
 
-    # Step 2: Generate image and caption
+    # Step 2: Get image + caption
     try:
-        image_caption = image_client.predict(
+        result = image_client.predict(
             prompt=topic,
             api_name="/generate_image_and_caption"
         )
+        if isinstance(result, (list, tuple)) and len(result) == 2:
+            image_url_or_path, caption = result
+        else:
+            image_url_or_path, caption = None, "⚠️ Unexpected image format."
     except Exception as e:
-        image_caption = f"Error generating image: {e}"
+        image_url_or_path, caption = None, f"❌ Error generating image: {e}"
 
-    # Step 3: Generate narrative continuation
+    # Step 3: Get story content
     try:
         gemini_narrative = chat_client.predict(
             user_input=f"Write a short story based on: {topic}",
             api_name="/chat_with_gemini"
         )
     except Exception as e:
-        gemini_narrative = f"Error with Gemini chat: {e}"
+        gemini_narrative = f"❌ Error generating narrative: {e}"
 
-    # Assemble the "book" content
-    book_text = f"""# Illustrated Book: "{topic.title()}"
-
-## Chapter 1: The Facts  
-{grounded_response}
-
-## Chapter 2: The Picture  
-🖼️ {image_caption}
-
-## Chapter 3: The Story  
-{gemini_narrative}
-"""
-    return book_text
+    return grounded_response, image_url_or_path, caption, gemini_narrative
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 📖 AI-Generated Illustrated Book")
-    topic = gr.Textbox(label="Enter a topic for your book", placeholder="e.g. Who won the euro 2024?")
-    output = gr.Textbox(label="Generated Book", lines=20)
-    button = gr.Button("Generate Book")
-
-    button.click(fn=generate_book, inputs=topic, outputs=output)
+    gr.Markdown("## 📘 AI-Generated Illustrated Book")
+
+    topic_input = gr.Textbox(label="Book Topic", placeholder="e.g. Who won Euro 2024?")
+    generate_btn = gr.Button("📚 Generate Book")
+
+    with gr.Column():
+        grounded_output = gr.Textbox(label="🌐 Chapter 1: Grounded Facts", lines=4)
+        image_output = gr.Image(label="🖼️ Chapter 2: Illustration")
+        caption_output = gr.Textbox(label="Image Caption")
+        narrative_output = gr.Textbox(label="✍️ Chapter 3: Story", lines=10)
+
+    generate_btn.click(
+        fn=generate_book,
+        inputs=topic_input,
+        outputs=[grounded_output, image_output, caption_output, narrative_output]
+    )
 
 demo.launch()