Update app.py

app.py

@@ -120,7 +120,38 @@ class CodingCopilot:
         
         return combined_content, status
 
-    def analyze_code(self, code_content: str, analysis_type: str, custom_prompt: str = "") -> str:
+    def analyze_code_with_chat(self, code_content: str, analysis_type: str, custom_prompt: str = "", chat_history: List = None) -> Tuple[str, List]:
+        """Analyze code with chat functionality."""
+        if chat_history is None:
+            chat_history = []
+        
+        # If this is the first analysis, do the standard analysis
+        if not chat_history:
+            result = self.analyze_code(code_content, analysis_type, custom_prompt)
+            chat_history.append(("Initial Analysis", result))
+            return result, chat_history
+        else:
+            # This is a follow-up question
+            context = ""
+            if chat_history:
+                context = "\n\nPrevious analysis conversation:\n"
+                for i, (q, a) in enumerate(chat_history[-2:]):  # Last 2 exchanges for context
+                    context += f"Q{i+1}: {q}\nA{i+1}: {a}\n"
+            
+            prompt = f"""Given this code:
+
+```python
+{code_content}
+```
+{context}
+
+Follow-up question about the analysis: {custom_prompt}
+
+Please provide a detailed response based on the code and previous analysis."""
+            
+            response = self.call_claude(prompt)
+            chat_history.append((custom_prompt, response))
+            return response, chat_history
         """Analyze code using Claude with optional custom prompt."""
         base_prompts = {
             "review": f"""Please review this code and provide:
@@ -364,17 +395,38 @@ custom_css = """
 }
 """
 
-def analyze_uploaded_code(files, analysis_type, custom_prompt):
-    """Analyze uploaded code files."""
+def analyze_uploaded_code(files, analysis_type, custom_prompt, chat_history):
+    """Analyze uploaded code files with chat functionality."""
     if not files:
-        return "Please upload at least one file."
+        return "Please upload at least one file.", chat_history, ""
     
     content, status = copilot.process_multiple_files(files)
     if not content or content.startswith("Error"):
-        return status or content
+        return status or content, chat_history, ""
+    
+    if chat_history is None:
+        chat_history = []
+    
+    # Check if this is initial analysis or follow-up chat
+    if not chat_history and analysis_type != "chat":
+        # Initial analysis
+        result = copilot.analyze_code(content, analysis_type, custom_prompt)
+        chat_history.append((f"Initial {analysis_type} analysis", result))
+        status_with_result = f"**Status:** {status}\n\n**Analysis Result:**\n{result}"
+    else:
+        # Follow-up chat or direct chat
+        if not custom_prompt.strip():
+            return "Please ask a question about the analyzed code.", chat_history, ""
+        
+        result, chat_history = copilot.analyze_code_with_chat(content, analysis_type, custom_prompt, chat_history)
+        status_with_result = result
+    
+    # Format chat history for display
+    chat_display = ""
+    for i, (q, a) in enumerate(chat_history):
+        chat_display += f"**Q{i+1}:** {q}\n\n**A{i+1}:** {a}\n\n---\n\n"
     
-    result = copilot.analyze_code(content, analysis_type, custom_prompt)
-    return f"**Status:** {status}\n\n**Analysis Result:**\n{result}"
+    return status_with_result, chat_history, chat_display
 
 def generate_new_code(description, language, reference_files):
     """Generate new code based on description with reference files."""
@@ -432,7 +484,7 @@ with gr.Blocks(title="🔥 Advanced Coding Copilot", css=custom_css, theme=gr.th
     with gr.Tabs():
         # Enhanced File Analysis Tab
         with gr.TabItem("📁 Analyze Code"):
-            gr.Markdown("### Upload multiple files and get comprehensive analysis")
+            gr.Markdown("### Upload multiple files and get comprehensive analysis with chat support")
             with gr.Row():
                 with gr.Column(scale=1):
                     file_input = gr.File(
@@ -442,34 +494,69 @@ with gr.Blocks(title="🔥 Advanced Coding Copilot", css=custom_css, theme=gr.th
                         height=120
                     )
                     analysis_type = gr.Dropdown(
-                        choices=["review", "explain", "optimize", "debug", "document", "custom"],
+                        choices=["review", "explain", "optimize", "debug", "document", "custom", "chat"],
                         value="review",
                         label="🔍 Analysis Type"
                     )
                     custom_prompt = gr.Textbox(
-                        label="✏️ Custom Prompt (when Analysis Type = 'custom')",
+                        label="✏️ Custom Prompt / Chat Message",
                         lines=3,
-                        placeholder="Write your custom analysis prompt here...",
+                        placeholder="Write your custom analysis prompt or ask questions about the code...",
                         visible=False
                     )
                     analyze_btn = gr.Button("🚀 Analyze Code", variant="primary", size="lg")
+                    clear_analysis_btn = gr.Button("🗑️ Clear Analysis Chat", variant="secondary")
                 
                 with gr.Column(scale=2):
                     analysis_output = gr.Textbox(
-                        label="📊 Analysis Result",
-                        lines=25,
-                        max_lines=40
+                        label="📊 Analysis Result / Current Response",
+                        lines=15,
+                        max_lines=25
+                    )
+                    analysis_chat_history = gr.Textbox(
+                        label="💬 Analysis Chat History",
+                        lines=15,
+                        max_lines=25,
+                        placeholder="Chat history will appear here after analysis..."
                     )
             
+            # Hidden state for analysis chat history
+            analysis_chat_state = gr.State([])
+            
             # Show/hide custom prompt based on analysis type
-            def toggle_custom_prompt(analysis_type):
-                return gr.update(visible=(analysis_type == "custom"))
+            def toggle_custom_prompt_analysis(analysis_type):
+                if analysis_type in ["custom", "chat"]:
+                    return gr.update(visible=True, label="✏️ Custom Prompt" if analysis_type == "custom" else "💬 Chat Message")
+                return gr.update(visible=False)
+            
+            def clear_analysis_chat():
+                return [], "", ""
+            
+            def update_button_text(analysis_type):
+                if analysis_type == "chat":
+                    return gr.update(value="💭 Send Message")
+                return gr.update(value="🚀 Analyze Code")
+            
+            analysis_type.change(
+                toggle_custom_prompt_analysis, 
+                inputs=analysis_type, 
+                outputs=custom_prompt
+            )
+            analysis_type.change(
+                update_button_text,
+                inputs=analysis_type,
+                outputs=analyze_btn
+            )
             
-            analysis_type.change(toggle_custom_prompt, inputs=analysis_type, outputs=custom_prompt)
             analyze_btn.click(
                 analyze_uploaded_code,
-                inputs=[file_input, analysis_type, custom_prompt],
-                outputs=analysis_output
+                inputs=[file_input, analysis_type, custom_prompt, analysis_chat_state],
+                outputs=[analysis_output, analysis_chat_state, analysis_chat_history]
+            )
+            
+            clear_analysis_btn.click(
+                clear_analysis_chat,
+                outputs=[analysis_chat_state, analysis_chat_history, analysis_output]
             )
         
         # Enhanced Code Generation Tab