commit

app.py

@@ -338,12 +338,25 @@ def query_agent(epub_file, prompt):
     else:
         return f"No plot image found. Raw response: {final_output}"
 
-gr.Interface(
-    fn=query_agent,
-    inputs=[
-        gr.File(label="Upload EPUB", type="filepath"),
-        gr.Textbox(label="Prompt", placeholder="e.g., Generate word frequency heatmap.")
-    ],
-    outputs=gr.Image(label="Sentiment Arc or Heatmap", type="filepath"),
-    title="Book Analyzer"
-).launch(share=True)
+with gr.Blocks() as demo:
+    gr.Markdown("### Chaptered - AI-Powered Novel Analyzer")
+    gr.Markdown("""
+    Upload your EPUB file and enter a prompt to analyze the novel.
+    You can ask for a word frequency heatmap, sentiment arc, or chapter-wise word count plot.
+    """)
+
+    with gr.Row():
+        with gr.Column():
+            epub_input = gr.File(label="Upload EPUB", type="filepath")
+            prompt_input = gr.Textbox(label="Prompt", placeholder="e.g., Generate word frequency heatmap.")
+            analyze_button = gr.Button("Analyze")
+        with gr.Column():
+            output_image = gr.Image(label="Sentiment Arc or Heatmap", type="filepath")
+
+    analyze_button.click(
+        fn=query_agent,
+        inputs=[epub_input, prompt_input],
+        outputs=output_image
+    )
+
+demo.launch(share=True)