Update app.py

app.py

@@ -29,8 +29,8 @@ class CodingCopilot:
             print(f"Error setting up AWS client: {e}")
             self.bedrock_client = None
 
-    def call_claude(self, prompt: str, max_tokens: int = 4000) -> str:
-        """Call Claude 3 Haiku via AWS Bedrock."""
+    def call_claude(self, prompt: str, max_tokens: int = 8000) -> str:
+        """Call Claude 3 Haiku via AWS Bedrock with increased token limit."""
         if not self.bedrock_client:
             return "Error: AWS Bedrock client not initialized. Please check your credentials."
         
@@ -121,7 +121,7 @@ class CodingCopilot:
         return combined_content, status
 
     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."""
+        """Analyze code with chat functionality and support for long responses."""
         if chat_history is None:
             chat_history = []
         
@@ -135,7 +135,7 @@ class CodingCopilot:
             context = ""
             if chat_history:
                 context = "\n\nPrevious analysis conversation:\n"
-                for i, (q, a) in enumerate(chat_history[-2:]):  # Last 2 exchanges for context
+                for i, (q, a) in enumerate(chat_history[-3:]):  # Last 3 exchanges for context
                     context += f"Q{i+1}: {q}\nA{i+1}: {a}\n"
             
             prompt = f"""Given this code:
@@ -147,11 +147,13 @@ class CodingCopilot:
 
 Follow-up question about the analysis: {custom_prompt}
 
-Please provide a detailed response based on the code and previous analysis."""
+Please provide a detailed response based on the code and previous analysis. If you're providing code, make sure to include the complete implementation without truncation."""
             
-            response = self.call_claude(prompt)
+            response = self.call_claude(prompt, max_tokens=8000)
             chat_history.append((custom_prompt, response))
             return response, chat_history
+
+    def analyze_code(self, code_content: str, analysis_type: str, custom_prompt: str = "") -> str:
         """Analyze code using Claude with optional custom prompt."""
         base_prompts = {
             "review": f"""Please review this code and provide:
@@ -159,35 +161,45 @@ Please provide a detailed response based on the code and previous analysis."""
 2. Potential bugs or issues
 3. Performance improvements
 4. Best practices suggestions
-5. Security considerations""",
+5. Security considerations
+
+Make sure to provide complete analysis with detailed explanations.""",
             
             "explain": f"""Please explain this code in detail:
 1. What does this code do?
 2. How does it work?
 3. Key functions and their purposes
 4. Dependencies and requirements
-5. Usage examples""",
+5. Usage examples
+
+Provide comprehensive explanations for each part.""",
             
             "optimize": f"""Please optimize this code:
 1. Suggest performance improvements
 2. Refactor for better readability
 3. Reduce complexity where possible
 4. Improve memory usage
-5. Provide the optimized version""",
+5. Provide the complete optimized version
+
+Make sure to include the full optimized code without truncation.""",
             
             "debug": f"""Please help debug this code:
 1. Identify potential bugs
 2. Suggest fixes
 3. Add error handling
 4. Improve robustness
-5. Provide corrected version""",
+5. Provide the complete corrected version
+
+Include the full corrected code implementation.""",
             
             "document": f"""Please add comprehensive documentation to this code:
 1. Add docstrings to functions/classes
 2. Add inline comments
 3. Create usage examples
 4. Document parameters and return values
-5. Add type hints where appropriate""",
+5. Add type hints where appropriate
+
+Provide the complete documented version of the code.""",
             
             "custom": custom_prompt
         }
@@ -204,9 +216,39 @@ Code:
 {code_content}
 ```
 
-Please provide a detailed response and if fixes are needed, provide the corrected code."""
+Please provide a detailed response and if fixes are needed, provide the complete corrected code without any truncation."""
+        
+        return self.call_claude(full_prompt, max_tokens=8000)
+
+    def generate_code_with_chat(self, description: str, language: str = "python", reference_files: str = "", chat_history: List = None) -> Tuple[str, List]:
+        """Generate code with chat functionality for continuation and modifications."""
+        if chat_history is None:
+            chat_history = []
         
-        return self.call_claude(full_prompt)
+        # If this is the first generation, do the standard generation
+        if not chat_history:
+            result = self.generate_code(description, language, reference_files)
+            chat_history.append(("Initial Generation Request", description))
+            chat_history.append(("Generated Code", result))
+            return result, chat_history
+        else:
+            # This is a follow-up request (continue, modify, etc.)
+            context = ""
+            if chat_history:
+                context = "\n\nPrevious generation conversation:\n"
+                for i, (q, a) in enumerate(chat_history[-4:]):  # Last 4 exchanges for context
+                    context += f"Step {i+1}: {q}\nResponse {i+1}: {a}\n"
+            
+            prompt = f"""Based on the previous code generation context:
+{context}
+
+New request: {description}
+
+Please provide the requested modification/continuation. If you're continuing code, make sure to provide the complete implementation. If you're modifying existing code, provide the full modified version."""
+            
+            response = self.call_claude(prompt, max_tokens=8000)
+            chat_history.append((description, response))
+            return response, chat_history
 
     def generate_code(self, description: str, language: str = "python", reference_files: str = "") -> str:
         """Generate code based on description with optional reference files."""
@@ -231,14 +273,16 @@ Requirements:
 3. Add type hints where appropriate
 4. Provide usage examples
 5. Follow best practices
+6. Generate complete, comprehensive code - don't truncate or abbreviate
+7. If the code is large, provide the full implementation
 {reference_context}
 
-Please provide the complete, ready-to-use code with detailed comments."""
+Please provide the complete, ready-to-use code with detailed comments. Make sure to generate comprehensive code that fully implements the requested functionality."""
         
-        return self.call_claude(prompt)
+        return self.call_claude(prompt, max_tokens=8000)
 
     def chat_with_code(self, code_content: str, question: str, chat_history: List = None) -> Tuple[str, List]:
-        """Chat about specific code with history."""
+        """Chat about specific code with history and support for long responses."""
         if chat_history is None:
             chat_history = []
         
@@ -258,9 +302,9 @@ Please provide the complete, ready-to-use code with detailed comments."""
 
 Current question: {question}
 
-Please provide a detailed answer about the code. If you're suggesting changes, provide the complete corrected code."""
+Please provide a detailed answer about the code. If you're suggesting changes or providing code, make sure to include the complete implementation without truncation."""
         
-        response = self.call_claude(prompt)
+        response = self.call_claude(prompt, max_tokens=8000)
         
         # Update chat history
         chat_history.append((question, response))
@@ -396,7 +440,7 @@ custom_css = """
 """
 
 def analyze_uploaded_code(files, analysis_type, custom_prompt, chat_history):
-    """Analyze uploaded code files with chat functionality."""
+    """Analyze uploaded code files with chat functionality and long response support."""
     if not files:
         return "Please upload at least one file.", chat_history, ""
     
@@ -428,34 +472,50 @@ def analyze_uploaded_code(files, analysis_type, custom_prompt, chat_history):
     
     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."""
+def generate_new_code_with_chat(description, language, reference_files, generation_chat_history):
+    """Generate new code with chat support for continuation and modifications."""
     if not description.strip():
-        return "Please provide a description of what you want to generate.", None
+        return "Please provide a description of what you want to generate.", None, generation_chat_history, ""
     
     reference_content = ""
     if reference_files:
         reference_content, _ = copilot.process_multiple_files(reference_files)
     
-    result = copilot.generate_code(description, language, reference_content)
+    if generation_chat_history is None:
+        generation_chat_history = []
+    
+    # Check if this is initial generation or continuation
+    if not generation_chat_history:
+        # Initial generation
+        result = copilot.generate_code(description, language, reference_content)
+        generation_chat_history.append(("Initial Generation", description))
+        generation_chat_history.append(("Generated Code", result))
+    else:
+        # Continuation or modification
+        result, generation_chat_history = copilot.generate_code_with_chat(description, language, reference_content, generation_chat_history)
     
     # Create downloadable file
     filename = f"generated_{language}_code.py"
     temp_file = copilot.create_downloadable_file(result, filename)
     
-    return result, temp_file
+    # Format chat history for display
+    chat_display = ""
+    for i, (q, a) in enumerate(generation_chat_history):
+        chat_display += f"**Step {i+1}:** {q}\n\n**Response {i+1}:** {a}\n\n---\n\n"
+    
+    return result, temp_file, generation_chat_history, chat_display
 
 def chat_about_code(files, question, chat_history):
-    """Chat about uploaded code with history."""
+    """Chat about uploaded code with history and long response support."""
     if not files:
-        return "Please upload at least one file.", chat_history, ""
+        return "Please upload at least one file.", chat_history, "", None
     
     if not question.strip():
-        return "Please ask a question about the code.", chat_history, ""
+        return "Please ask a question about the code.", chat_history, "", None
     
     content, status = copilot.process_multiple_files(files)
     if not content or content.startswith("Error"):
-        return status or content, chat_history, ""
+        return status or content, chat_history, "", None
     
     if chat_history is None:
         chat_history = []
@@ -475,10 +535,11 @@ def chat_about_code(files, question, chat_history):
     return response, updated_history, chat_display, download_file
 
 # Create Gradio interface with custom theme
-with gr.Blocks(title="🔥 Advanced Coding Copilot", css=custom_css, theme=gr.themes.Base()) as app:
+with gr.Blocks(title="🔥 Advanced Coding Copilot - Long Code Support", css=custom_css, theme=gr.themes.Base()) as app:
     gr.Markdown("""
-    # 🔥 Advanced Coding Copilot
+    # 🔥 Advanced Coding Copilot - Long Code Support
     ### Upload multiple Python files (.py) or Jupyter notebooks (.ipynb) for intelligent analysis, or generate new code with AI assistance!
+    ### ✨ **NEW**: Support for extremely long code generation (up to 5000+ lines) with chat continuation!
     """)
     
     with gr.Tabs():
@@ -501,7 +562,7 @@ with gr.Blocks(title="🔥 Advanced Coding Copilot", css=custom_css, theme=gr.th
                     custom_prompt = gr.Textbox(
                         label="✏️ Custom Prompt / Chat Message",
                         lines=3,
-                        placeholder="Write your custom analysis prompt or ask questions about the code...",
+                        placeholder="Write your custom analysis prompt, ask questions, or request 'continue' for more details...",
                         visible=False
                     )
                     analyze_btn = gr.Button("🚀 Analyze Code", variant="primary", size="lg")
@@ -511,12 +572,12 @@ with gr.Blocks(title="🔥 Advanced Coding Copilot", css=custom_css, theme=gr.th
                     analysis_output = gr.Textbox(
                         label="📊 Analysis Result / Current Response",
                         lines=15,
-                        max_lines=25
+                        max_lines=30
                     )
                     analysis_chat_history = gr.Textbox(
                         label="💬 Analysis Chat History",
                         lines=15,
-                        max_lines=25,
+                        max_lines=30,
                         placeholder="Chat history will appear here after analysis..."
                     )
             
@@ -559,15 +620,15 @@ with gr.Blocks(title="🔥 Advanced Coding Copilot", css=custom_css, theme=gr.th
                 outputs=[analysis_chat_state, analysis_chat_history, analysis_output]
             )
         
-        # Enhanced Code Generation Tab
+        # Enhanced Code Generation Tab with Chat Support
         with gr.TabItem("⚡ Generate Code"):
-            gr.Markdown("### Generate new code with optional reference files")
+            gr.Markdown("### Generate new code with chat support for continuation and modifications")
             with gr.Row():
                 with gr.Column(scale=1):
                     code_description = gr.Textbox(
-                        label="📝 Describe what you want to code",
+                        label="📝 Describe what you want to code / Chat Message",
                         lines=5,
-                        placeholder="E.g., Create a REST API using FastAPI with authentication, database models, and CRUD operations..."
+                        placeholder="E.g., Create a REST API using FastAPI... OR 'continue the previous code' OR 'add more features' OR 'optimize the generated code'"
                     )
                     code_language = gr.Dropdown(
                         choices=["python", "javascript", "java", "cpp", "c", "go", "rust"],
@@ -579,13 +640,21 @@ with gr.Blocks(title="🔥 Advanced Coding Copilot", css=custom_css, theme=gr.th
                         file_count="multiple",
                         file_types=[".py", ".ipynb"]
                     )
-                    generate_btn = gr.Button("🎯 Generate Code", variant="primary", size="lg")
+                    generate_btn = gr.Button("🎯 Generate/Continue Code", variant="primary", size="lg")
+                    clear_generation_btn = gr.Button("🗑️ Clear Generation Chat", variant="secondary")
                 
                 with gr.Column(scale=2):
                     generated_code = gr.Code(
                         label="🔧 Generated Code",
                         language="python",
-                        lines=25
+                        lines=25,
+                        max_lines=50
+                    )
+                    generation_chat_history = gr.Textbox(
+                        label="💬 Generation Chat History",
+                        lines=10,
+                        max_lines=20,
+                        placeholder="Generation history will appear here..."
                     )
                     download_generated = gr.File(
                         label="💾 Download Generated Code",
@@ -593,19 +662,26 @@ with gr.Blocks(title="🔥 Advanced Coding Copilot", css=custom_css, theme=gr.th
                         interactive=False
                     )
             
-            def handle_generation(desc, lang, ref_files):
-                code, file_path = generate_new_code(desc, lang, ref_files)
-                return code, gr.update(value=file_path, visible=True)
+            # Hidden state for generation chat history
+            generation_chat_state = gr.State([])
+            
+            def clear_generation_chat():
+                return [], "", ""
             
             generate_btn.click(
-                handle_generation,
-                inputs=[code_description, code_language, reference_files],
-                outputs=[generated_code, download_generated]
+                generate_new_code_with_chat,
+                inputs=[code_description, code_language, reference_files, generation_chat_state],
+                outputs=[generated_code, download_generated, generation_chat_state, generation_chat_history]
+            )
+            
+            clear_generation_btn.click(
+                clear_generation_chat,
+                outputs=[generation_chat_state, generation_chat_history, generated_code]
             )
         
         # Enhanced Code Chat Tab
         with gr.TabItem("💬 Chat About Code"):
-            gr.Markdown("### Interactive chat about your code with conversation history")
+            gr.Markdown("### Interactive chat about your code with conversation history and long response support")
             with gr.Row():
                 with gr.Column(scale=1):
                     chat_file_input = gr.File(
@@ -617,7 +693,7 @@ with gr.Blocks(title="🔥 Advanced Coding Copilot", css=custom_css, theme=gr.th
                     code_question = gr.Textbox(
                         label="❓ Ask a question about your code",
                         lines=3,
-                        placeholder="E.g., How can I optimize this algorithm? What design patterns are used here?"
+                        placeholder="E.g., How can I optimize this algorithm? What design patterns are used here? Continue explaining... Add more features..."
                     )
                     chat_btn = gr.Button("💭 Ask Question", variant="primary", size="lg")
                     clear_chat_btn = gr.Button("🗑️ Clear Chat History", variant="secondary")
@@ -626,12 +702,12 @@ with gr.Blocks(title="🔥 Advanced Coding Copilot", css=custom_css, theme=gr.th
                     chat_output = gr.Textbox(
                         label="🤖 AI Response",
                         lines=15,
-                        max_lines=25
+                        max_lines=30
                     )
                     chat_history_display = gr.Textbox(
                         label="📜 Chat History",
                         lines=10,
-                        max_lines=20
+                        max_lines=25
                     )
                     download_chat = gr.File(
                         label="💾 Download Chat Response",
@@ -671,13 +747,21 @@ with gr.Blocks(title="🔥 Advanced Coding Copilot", css=custom_css, theme=gr.th
     gr.Markdown("""
     ---
     ### 🔥 Features:
+    - **🚀 Long Code Support**: Generate up to 5000+ lines of code with continuation support
+    - **💬 Chat Continuation**: Ask "continue", "add more features", or "extend the code" in any tab
     - **Multi-file Support**: Upload and analyze multiple files simultaneously
     - **Custom Prompts**: Write your own analysis prompts for specific needs
     - **Interactive Chat**: Have conversations about your code with full history
     - **File Downloads**: Get generated code and chat responses as downloadable files
     - **Reference Context**: Use existing files as reference when generating new code
     
-    **Powered by:** Claude 3 Haiku via AWS Bedrock | **Theme:** Cyberpunk Black & Red
+    ### 💡 Pro Tips:
+    - Use "continue" or "continue the code" to extend generated code
+    - Ask "add more features" to enhance existing implementations
+    - Use "optimize" or "refactor" to improve code quality
+    - Upload reference files to generate code based on existing patterns
+    
+    **Powered by:** Claude 3 Haiku via AWS Bedrock | **Theme:** Cyberpunk Black & Red | **Max Tokens:** 8000 per response
     """)
 
 # Launch the app