Update app.py

app.py

@@ -41,50 +41,51 @@ def generate_full_story_chunks(excerpt):
 def clear_history():
     return gr.Textbox.update(value="")
 
-# Build UI with Gradio Blocks
-gr.Markdown("## 🏺 Mythology Storyteller")
-gr.Markdown(
-    "Enter a phrase from a chapter of your choice "
-    "(please include Parv, key event, and section for better results)."
-)
+# Build UI inside Blocks
+with gr.Blocks() as demo:
+    gr.Markdown("## 🏺 Mythology Storyteller")
+    gr.Markdown(
+        "Enter a phrase from a chapter of your choice "
+        "(please include Parv, key event, and section for better results)."
+    )
 
-with gr.Row():
-    with gr.Column():
-        user_input = gr.Textbox(
-            label="Incomplete story excerpt",
-            placeholder="Enter an excerpt from the Mahabharata here...",
-            lines=4,
-        )
-        summary_input = gr.Textbox(
-            label="Chapter summary (optional)",
-            placeholder="Enter summary if available...",
-            lines=2,
-        )
-        generate_btn = gr.Button("✨ Generate Story")
+    with gr.Row():
+        with gr.Column():
+            user_input = gr.Textbox(
+                label="Incomplete story excerpt",
+                placeholder="Enter an excerpt from the Mahabharata here...",
+                lines=4,
+            )
+            summary_input = gr.Textbox(
+                label="Chapter summary (optional)",
+                placeholder="Enter summary if available...",
+                lines=2,
+            )
+            generate_btn = gr.Button("✨ Generate Story")
 
-    with gr.Column():
-        output_text = gr.Textbox(
-            label="Generated Story",
-            placeholder="Your generated story will appear here...",
-            lines=12,
-        )
-        clear_btn = gr.Button("🗑️ Clear Conversation")
+        with gr.Column():
+            output_text = gr.Textbox(
+                label="Generated Story",
+                placeholder="Your generated story will appear here...",
+                lines=12,
+            )
+            clear_btn = gr.Button("🗑️ Clear Conversation")
 
-gr.Markdown("---")
-gr.Markdown("🔌 Use via API (see Hugging Face Inference docs).")
+    gr.Markdown("---")
+    gr.Markdown("🔌 Use via API (see Hugging Face Inference docs).")
 
-# Button wiring with streaming-safe generator
-generate_btn.click(
-    fn=generate_full_story_chunks,
-    inputs=user_input,
-    outputs=output_text,
-)
+    # Button wiring inside the Blocks context
+    generate_btn.click(
+        fn=generate_full_story_chunks,
+        inputs=user_input,
+        outputs=output_text,
+    )
 
-clear_btn.click(
-    fn=clear_history,
-    inputs=None,
-    outputs=output_text,
-)
+    clear_btn.click(
+        fn=clear_history,
+        inputs=None,
+        outputs=output_text,
+    )
 
 # Launch app
 if __name__ == "__main__":