Update app.py

app.py

@@ -202,6 +202,24 @@ def upload_file_handler(files):
         return files
     return []
 
+def generate_plan(history, file_cache):
+    """Generate a plan using the planning prompt and Gemini API"""
+    
+    # Build conversation history
+    conversation_history = ""
+    if history:
+        for user_msg, ai_msg in history:
+            conversation_history += f"User: {user_msg}\n"
+            if ai_msg:
+                conversation_history += f"Assistant: {ai_msg}\n"
+    
+    # Format the prompt
+    formatted_prompt = hf_query_gen_prompt + "\n\n" + conversation_history
+    
+    # Get plan from Gemini
+    plan = generate_with_gemini(formatted_prompt, "Planning with gemini")
+    return plan
+
 # Custom CSS for a sleek design
 custom_css = """
 .gradio-container {
@@ -299,6 +317,14 @@ with gr.Blocks(css=custom_css, title="Data Science Requirements Gathering Agent"
                 bubble_full_width=False,
                 elem_classes=["chat-container"]
             )
+
+            plan_output = gr.Textbox(
+                            label="Generated Plan",
+                            interactive=False,
+                            visible=True,
+                            lines=10,
+                            max_lines=20
+                        )
             
             with gr.Row():
                 with gr.Column(scale=4):
@@ -310,6 +336,10 @@ with gr.Blocks(css=custom_css, title="Data Science Requirements Gathering Agent"
                     )
                 with gr.Column(scale=1):
                     send_btn = gr.Button("Send 📤", variant="primary", elem_classes=["btn-primary"])
+
+                with gr.Column(scale=1):
+                    plan_btn = gr.Button("Generate Plan 📋", variant="secondary", elem_classes=["btn-secondary"])
+
             
             with gr.Row():
                 clear_btn = gr.Button("Clear Chat 🗑️", variant="secondary", elem_classes=["btn-secondary"])
@@ -377,6 +407,12 @@ with gr.Blocks(css=custom_css, title="Data Science Requirements Gathering Agent"
         clear_chat,
         outputs=[chatbot, chat_history, file_cache]
     )
+
+    plan_btn.click(
+        generate_plan,
+        inputs=[chat_history, file_cache],
+        outputs=[plan_output]
+    )
     
     file_upload.change(
         lambda files: files,