Update app.py

app.py

@@ -3,7 +3,6 @@ import google.generativeai as genai
 import requests
 import subprocess
 import os
-import pylint
 import pandas as pd
 import numpy as np
 from sklearn.model_selection import train_test_split
@@ -67,15 +66,18 @@ def generate_response(user_input):
 
 def optimize_code(code):
     # Use abstract syntax tree for advanced code analysis
-    tree = ast.parse(code)
-    analyzer = CodeAnalyzer()
-    analyzer.visit(tree)
-    
-    # Apply code transformations based on analysis
-    transformer = CodeTransformer(analyzer.get_optimizations())
-    optimized_tree = transformer.visit(tree)
-    
-    optimized_code = ast.unparse(optimized_tree)
+    try:
+        tree = ast.parse(code)
+        analyzer = CodeAnalyzer()
+        analyzer.visit(tree)
+        
+        # Apply code transformations based on analysis
+        transformer = CodeTransformer(analyzer.get_optimizations())
+        optimized_tree = transformer.visit(tree)
+        
+        optimized_code = ast.unparse(optimized_tree)
+    except SyntaxError as e:
+        return code, f"SyntaxError: {str(e)}"
     
     # Run pylint for additional suggestions
     with open("temp_code.py", "w") as file:
@@ -122,31 +124,119 @@ def analyze_code_quality(code):
     return prediction.item()  # Return the probability of needing improvement
 
 def visualize_code_structure(code):
-    tree = ast.parse(code)
-    graph = nx.DiGraph()
-    
-    def add_nodes_edges(node, parent=None):
-        node_id = id(node)
-        graph.add_node(node_id, label=type(node).__name__)
-        if parent:
-            graph.add_edge(id(parent), node_id)
-        for child in ast.iter_child_nodes(node):
-            add_nodes_edges(child, node)
-    
-    add_nodes_edges(tree)
-    
-    plt.figure(figsize=(12, 8))
-    pos = nx.spring_layout(graph)
-    nx.draw(graph, pos, with_labels=True, node_color='lightblue', node_size=1000, font_size=8, font_weight='bold')
-    labels = nx.get_node_attributes(graph, 'label')
-    nx.draw_networkx_labels(graph, pos, labels, font_size=6)
-    
-    return plt
+    try:
+        tree = ast.parse(code)
+        graph = nx.DiGraph()
+        
+        def add_nodes_edges(node, parent=None):
+            node_id = id(node)
+            graph.add_node(node_id, label=type(node).__name__)
+            if parent:
+                graph.add_edge(id(parent), node_id)
+            for child in ast.iter_child_nodes(node):
+                add_nodes_edges(child, node)
+        
+        add_nodes_edges(tree)
+        
+        plt.figure(figsize=(12, 8))
+        pos = nx.spring_layout(graph)
+        nx.draw(graph, pos, with_labels=True, node_color='lightblue', node_size=1000, font_size=8, font_weight='bold')
+        labels = nx.get_node_attributes(graph, 'label')
+        nx.draw_networkx_labels(graph, pos, labels, font_size=6)
+        
+        return plt
+    except SyntaxError:
+        return None
 
 # Streamlit UI setup
 st.set_page_config(page_title="Advanced AI Code Assistant", page_icon="🚀", layout="wide")
 
-# ... (keep the existing CSS styles) ...
+st.markdown("""
+<style>
+    @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;600;700&display=swap');
+    
+    body {
+        font-family: 'Inter', sans-serif;
+        background-color: #f0f4f8;
+        color: #1a202c;
+    }
+    .stApp {
+        max-width: 1200px;
+        margin: 0 auto;
+        padding: 2rem;
+    }
+    .main-container {
+        background: #ffffff;
+        border-radius: 16px;
+        padding: 2rem;
+        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.05);
+    }
+    h1 {
+        font-size: 2.5rem;
+        font-weight: 700;
+        color: #2d3748;
+        text-align: center;
+        margin-bottom: 1rem;
+    }
+    .subtitle {
+        font-size: 1.1rem;
+        text-align: center;
+        color: #4a5568;
+        margin-bottom: 2rem;
+    }
+    .stTextArea textarea {
+        border: 2px solid #e2e8f0;
+        border-radius: 8px;
+        font-size: 1rem;
+        padding: 0.75rem;
+        transition: all 0.3s ease;
+    }
+    .stTextArea textarea:focus {
+        border-color: #4299e1;
+        box-shadow: 0 0 0 3px rgba(66, 153, 225, 0.5);
+    }
+    .stButton button {
+        background-color: #4299e1;
+        color: white;
+        border: none;
+        border-radius: 8px;
+        font-size: 1.1rem;
+        font-weight: 600;
+        padding: 0.75rem 2rem;
+        transition: all 0.3s ease;
+        width: 100%;
+    }
+    .stButton button:hover {
+        background-color: #3182ce;
+    }
+    .output-container {
+        background: #f7fafc;
+        border-radius: 8px;
+        padding: 1rem;
+        margin-top: 2rem;
+    }
+    .code-block {
+        background-color: #2d3748;
+        color: #e2e8f0;
+        font-family: 'Fira Code', monospace;
+        font-size: 0.9rem;
+        border-radius: 8px;
+        padding: 1rem;
+        margin-top: 1rem;
+        overflow-x: auto;
+    }
+    .stAlert {
+        background-color: #ebf8ff;
+        color: #2b6cb0;
+        border-radius: 8px;
+        border: none;
+        padding: 0.75rem 1rem;
+    }
+    .stSpinner {
+        color: #4299e1;
+    }
+</style>
+""", unsafe_allow_html=True)
 
 st.markdown('<div class="main-container">', unsafe_allow_html=True)
 st.title("🚀 Advanced AI Code Assistant")
@@ -164,27 +254,35 @@ if st.button("Generate Advanced Code"):
                 st.error(completed_text)
             else:
                 optimized_code, lint_results = optimize_code(completed_text)
-                quality_score = analyze_code_quality(optimized_code)
-                
-                st.success(f"Code generated and optimized successfully! Quality Score: {quality_score:.2f}")
-                
-                st.markdown('<div class="output-container">', unsafe_allow_html=True)
-                st.markdown('<div class="code-block">', unsafe_allow_html=True)
-                st.code(optimized_code)
-                st.markdown('</div>', unsafe_allow_html=True)
-                
-                with st.expander("View Code Structure Visualization"):
-                    st.pyplot(visualize_code_structure(optimized_code))
-                
-                with st.expander("View Lint Results"):
-                    st.text(lint_results)
-                
-                with st.expander("Fetch Similar Code from GitHub"):
-                    github_results = fetch_from_github(prompt)
-                    for item in github_results:
-                        st.markdown(f"[{item['name']}]({item['html_url']})")
                 
-                st.markdown('</div>', unsafe_allow_html=True)
+                if "SyntaxError" in lint_results:
+                    st.warning(f"Syntax error detected: {lint_results}")
+                    st.code(completed_text)
+                else:
+                    quality_score = analyze_code_quality(optimized_code)
+                    st.success(f"Code generated and optimized successfully! Quality Score: {quality_score:.2f}")
+                    
+                    st.markdown('<div class="output-container">', unsafe_allow_html=True)
+                    st.markdown('<div class="code-block">', unsafe_allow_html=True)
+                    st.code(optimized_code)
+                    st.markdown('</div>', unsafe_allow_html=True)
+                    
+                    visualization = visualize_code_structure(optimized_code)
+                    if visualization:
+                        with st.expander("View Code Structure Visualization"):
+                            st.pyplot(visualization)
+                    else:
+                        st.warning("Unable to generate code structure visualization due to syntax errors.")
+                    
+                    with st.expander("View Lint Results"):
+                        st.text(lint_results)
+                    
+                    with st.expander("Fetch Similar Code from GitHub"):
+                        github_results = fetch_from_github(prompt)
+                        for item in github_results:
+                            st.markdown(f"[{item['name']}]({item['html_url']})")
+                    
+                    st.markdown('</div>', unsafe_allow_html=True)
 
 st.markdown("""
 <div style='text-align: center; margin-top: 2rem; color: #4a5568;'>