bug fix

app.py

@@ -188,6 +188,58 @@ def create_connected_mode_ui():
 
     return demo
 
+def create_document_analysis_ui():
+    """Creates the Gradio UI for the Document Analysis Mode."""
+    with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple", secondary_hue="pink")) as demo:
+        gr.Markdown("# 🌽 Aura Mind Glow: Document Analysis Mode 📄")
+        gr.Markdown("Upload a PDF or a spreadsheet and ask questions about its content.")
+
+        with gr.Row():
+            with gr.Column(scale=1):
+                doc_input = gr.File(label="Upload Document", file_types=[".pdf", ".csv"])
+                query_input = gr.Textbox(label="Ask a question about the document")
+                submit_btn = gr.Button("Analyze and Query")
+            with gr.Column(scale=2):
+                answer_output = gr.Textbox(label="Answer", interactive=False, lines=10)
+                status_output = gr.Textbox(label="Status", interactive=False, lines=3)
+
+        def analysis_process(doc, query):
+            if doc is None:
+                yield "Please upload a document to begin.", ""
+                return
+
+            file_path = doc.name
+            file_ext = os.path.splitext(file_path)[1].lower()
+
+            if file_ext == ".pdf":
+                yield "Analyzing PDF...", ""
+                chain, vector_store = analyze_pdf(file_path)
+                if chain and vector_store:
+                    yield "PDF analyzed successfully. Now querying...", ""
+                    answer = query_pdf(chain, vector_store, query)
+                    yield answer, "Query complete."
+                else:
+                    yield "Failed to analyze PDF.", "Error"
+            elif file_ext == ".csv":
+                yield "Analyzing spreadsheet...", ""
+                agent = analyze_spreadsheet(file_path)
+                if agent:
+                    yield "Spreadsheet analyzed successfully. Now querying...", ""
+                    answer = query_spreadsheet(agent, query)
+                    yield answer, "Query complete."
+                else:
+                    yield "Failed to analyze spreadsheet.", "Error"
+            else:
+                yield "Unsupported file type. Please upload a PDF or a CSV file.", "Error"
+
+        submit_btn.click(
+            analysis_process,
+            inputs=[doc_input, query_input],
+            outputs=[answer_output, status_output]
+        )
+    return demo
+
+
 def create_story_mode_ui():
     """Creates the Gradio UI for the Farmer's Story Mode."""
     with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="yellow")) as demo:
@@ -260,12 +312,18 @@ if __name__ == "__main__":
         else:
             print("⚠️ Connected Mode disabled: ADK components not initialized.")
 
-                # Add the new Document Analysis UI
+        if STORY_LLM:
+            story_mode_ui = create_story_mode_ui()
+            interface_list.append(story_mode_ui)
+            tab_titles.append("Farmer's Story Mode")
+        else:
+            print("⚠️ Farmer's Story Mode disabled: Story LLM not initialized.")
+
+        # Add the new Document Analysis UI
         document_analysis_ui = create_document_analysis_ui()
         interface_list.append(document_analysis_ui)
         tab_titles.append("Document Analysis")
-        else:
-            print("⚠️ Farmer's Story Mode disabled: Story LLM not initialized.")
+
     else:
         print("❌ No internet connection. Launching in Offline Mode only.")