Update app.py

app.py

@@ -46,49 +46,27 @@ 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': 0,
-            'class_def': 0,
-            'loop': 0,
-            'conditional': 0
+            '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 ']))
         }
-        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"""
         suggestions = []
-
-        if patterns['function_def'] >= 1:
-            suggestions.append("🔍 Consider organizing related functions into classes.")
-    
-        if patterns['loop'] >= 1:
-            suggestions.append("🔄 Try using list comprehensions or built-in functions like `map()` where appropriate.")
-    
-        if patterns['conditional'] >= 2:
-            suggestions.append("⚖ Consider simplifying conditionals using design patterns or helper functions.")
-    
-        if sum(patterns.values()) == 0:
-            suggestions.append("🧐 No recognizable code patterns found. Try pasting a complete code block.")
-    
-        return "\n".join(suggestions)
-
+        if patterns['function_def'] > 3:
+            suggestions.append("🔍 Consider breaking down into smaller functions or using a class structure.")
+        if patterns['loop'] > 2:
+            suggestions.append("🔄 You might benefit from list comprehensions or map/filter functions.")
+        if patterns['conditional'] > 3:
+            suggestions.append("⚖ Complex conditionals might be simplified using polymorphism or strategy pattern.")
+        return "\n".join(suggestions) if suggestions else "No specific suggestions at this time."
     
     def process_input(self, user_input):
         """Process user input and generate response"""
@@ -144,7 +122,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="🤖 AI Code Copilot") as demo:
             output_text = gr.Markdown()
             # Display suggestions
             gr.Markdown("**Suggestions:**")
-            suggestions = gr.Markdown()
+            suggestions_output = gr.Markdown()
             # Display pattern analysis
             gr.Markdown("**Pattern Analysis:**")
             pattern_display = gr.Dataframe(
@@ -155,7 +133,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="🤖 AI Code Copilot") as demo:
             )
 
     # Processing function for both button and enter
-    def process_input(user_input):
+    def process_input_wrapper(user_input):
         response, patterns, sugg = copilot.process_input(user_input)
         pattern_df = pd.DataFrame({
             "Pattern": list(patterns.keys()),
@@ -164,14 +142,14 @@ with gr.Blocks(theme=gr.themes.Soft(), title="🤖 AI Code Copilot") as demo:
         return response, sugg, pattern_df
     
     submit_btn.click(
-        fn=process_input,
+        fn=process_input_wrapper,
         inputs=input_text,
-        outputs=[output_text, suggestions, pattern_display]
+        outputs=[output_text, suggestions_output, pattern_display]
     )
     input_text.submit(
-        fn=process_input,
+        fn=process_input_wrapper,
         inputs=input_text,
-        outputs=[output_text, suggestions, pattern_display]
+        outputs=[output_text, suggestions_output, pattern_display]
     )
 
 if __name__ == "__main__":