Update app.py

app.py

@@ -5,7 +5,6 @@ from dotenv import load_dotenv
 import nltk
 from nltk.tokenize import sent_tokenize
 import pandas as pd
-import json
 
 # Initialize NLTK
 nltk.download('punkt', quiet=True)
@@ -29,7 +28,6 @@ class CodeCopilot:
         }
         
         try:
-            # Correct API endpoint based on Blackbox documentation
             response = requests.post(
                 "https://api.blackbox.ai/chat/completions",
                 headers=headers,
@@ -48,16 +46,31 @@ class CodeCopilot:
         except Exception as e:
             return f"Processing Error: {str(e)}"
     
+    import ast
+
     def analyze_code_patterns(self, text):
-        """Analyze text for coding patterns"""
-        sentences = sent_tokenize(text)
         patterns = {
-            'function_def': sum(1 for s in sentences if 'def ' in s),
-            'class_def': sum(1 for s in sentences if 'class ' in s),
-            'loop': sum(1 for s in sentences if any(word in s for word in ['for ', 'while ', 'loop'])),
-            'conditional': sum(1 for s in sentences if any(word in s for word in ['if ', 'else ', 'elif ']))
+            'function_def': 0,
+            'class_def': 0,
+            'loop': 0,
+            'conditional': 0
         }
+        try:
+            tree = ast.parse(text)
+            for node in ast.walk(tree):
+                if isinstance(node, ast.FunctionDef):
+                    patterns['function_def'] += 1
+                elif isinstance(node, ast.ClassDef):
+                    patterns['class_def'] += 1
+                elif isinstance(node, (ast.For, ast.While)):
+                    patterns['loop'] += 1
+                elif isinstance(node, ast.If):
+                    patterns['conditional'] += 1
+        except Exception as e:
+            print(f"AST parse error: {e}")
+        
         return patterns
+
     
     def generate_suggestions(self, patterns):
         """Generate suggestions based on detected patterns"""
@@ -72,10 +85,8 @@ class CodeCopilot:
     
     def process_input(self, user_input):
         """Process user input and generate response"""
-        # Analyze patterns
         patterns = self.analyze_code_patterns(user_input)
         
-        # Create context-aware prompt
         context = "\nPrevious conversation:\n" + "\n".join(
             [f"User: {h[0]}\nAI: {h[1]}" for h in self.chat_history[-self.context_window:]])
         
@@ -87,11 +98,9 @@ class CodeCopilot:
         {user_input}
         """
         
-        # Get response
         response = self.get_blackbox_response(prompt)
         suggestions = self.generate_suggestions(patterns)
         
-        # Update chat history
         self.chat_history.append((user_input, response))
         
         return response, patterns, suggestions
@@ -100,30 +109,45 @@ class CodeCopilot:
 copilot = CodeCopilot()
 
 # Gradio interface
-with gr.Blocks(theme=gr.themes.Soft(), title="AI Code Copilot") as demo:
-    gr.Markdown("""<h1 style="text-align: center">🤖 AI Code Copilot</h1>""")
-    
+with gr.Blocks(theme=gr.themes.Soft(), title="🤖 AI Code Copilot") as demo:
+    # Header
+    gr.Markdown(
+        """
+        <div style='text-align: center; margin-bottom: 1rem;'>
+            <h1>🤖 AI Code Copilot</h1>
+            <p>Your interactive assistant for code suggestions and analysis</p>
+        </div>
+        """
+    )
+
+    # Main layout: input on the left, outputs on the right
     with gr.Row():
-        with gr.Column(scale=3):
+        with gr.Column(scale=3, min_width=300):
             input_text = gr.Textbox(
-                label="Your Code or Question",
-                placeholder="Paste your code or ask a question...",
-                lines=7
+                label="Enter your code or question",
+                placeholder="Paste code snippets or ask a coding question...",
+                lines=10,
+                interactive=True
             )
-            submit_btn = gr.Button("Generate", variant="primary")
-            
-        with gr.Column(scale=7):
-            with gr.Tab("Assistant Response"):
-                output_text = gr.Markdown()
-            with gr.Tab("Suggestions"):
-                suggestions = gr.Markdown()
-            with gr.Tab("Pattern Analysis"):
-                pattern_display = gr.Dataframe(
-                    headers=["Pattern", "Count"],
-                    datatype=["str", "number"],
-                    interactive=False
-                )
-    
+            submit_btn = gr.Button("🚀 Generate", variant="primary")
+        
+        with gr.Column(scale=5, min_width=500):
+            # Display response
+            gr.Markdown("**Assistant Response:**")
+            output_text = gr.Markdown()
+            # Display suggestions
+            gr.Markdown("**Suggestions:**")
+            suggestions = gr.Markdown()
+            # Display pattern analysis
+            gr.Markdown("**Pattern Analysis:**")
+            pattern_display = gr.Dataframe(
+                headers=["Pattern", "Count"],
+                datatype=["str", "number"],
+                interactive=False,
+                label="Detected code patterns"
+            )
+
+    # Processing function for both button and enter
     def process_input(user_input):
         response, patterns, sugg = copilot.process_input(user_input)
         pattern_df = pd.DataFrame({
@@ -137,7 +161,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI Code Copilot") as demo:
         inputs=input_text,
         outputs=[output_text, suggestions, pattern_display]
     )
-    
     input_text.submit(
         fn=process_input,
         inputs=input_text,