Update app.py

app.py

@@ -5,7 +5,7 @@ import tempfile
 import uuid
 from dotenv import load_dotenv
 from langchain_community.vectorstores import FAISS
-from langchain_community.embeddings import HuggingFaceInstructEmbeddings, HuggingFaceEmbeddings
+from langchain_community.embeddings import HuggingFaceInstructEmbeddings
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 import fitz  # PyMuPDF
 import base64
@@ -38,8 +38,7 @@ except Exception as e:
         )
     except Exception as e:
         print(f"Warning: Failed to load fallback embeddings model: {e}")
-        # Use a simpler embeddings model as a fallback
-        embeddings = HuggingFaceEmbeddings()
+        embeddings = None
 
 # Directory to store FAISS indexes
 FAISS_INDEX_DIR = "faiss_indexes_tech"
@@ -49,54 +48,69 @@ if not os.path.exists(FAISS_INDEX_DIR):
 # Dictionary to store user-specific vectorstores
 user_vectorstores = {}
 
-# Custom CSS for Tech theme with modern UI enhancements
+# Custom CSS for Tech theme
 custom_css = """
-@import url('shared-base.css');
-
-/* Tech Vision Specific */
-.code-block {
-    background: var(--surface-2);
-    border-radius: var(--radius-md);
-    padding: var(--spacing-md);
-    color: var(--text-color);
-    font-family: 'JetBrains Mono', monospace;
-    margin: var(--spacing-md) 0;
-    border-left: 3px solid var(--accent-color);
-    box-shadow: var(--shadow-sm);
+:root {
+    --primary-color: #4285F4;
+    --secondary-color: #34A853;
+    --accent-color: #EA4335;
+    --light-background: #F8F9FA;
+    --dark-text: #202124;
+    --white: #FFFFFF;
+    --border-color: #DADCE0;
+    --code-bg: #F1F3F4;
 }
-
-.stats-card {
-    background: var(--surface-2);
-    color: var(--text-color);
-    padding: var(--spacing-md);
-    border-radius: var(--radius-md);
-    margin: var(--spacing-md) 0;
-    border: 1px solid var(--border-color);
-    box-shadow: var(--shadow-sm);
+body { 
+    background-color: var(--light-background); 
+    font-family: 'Google Sans', 'Roboto', sans-serif; 
 }
-
-/* Tech-specific accents */
-.function-name { color: var(--accent-color); }
-.keyword { color: var(--accent-color-alt); }
-.string { color: var(--text-color-secondary); }
-
-/* Custom Scrollbar */
-::-webkit-scrollbar {
-    width: 8px;
-    height: 8px;
+.container { 
+    max-width: 1200px !important; 
+    margin: 0 auto !important; 
+    padding: 10px; 
 }
-
-::-webkit-scrollbar-track {
-    background: var(--surface-1);
+.header { 
+    background-color: var(--white);
+    border-bottom: 1px solid var(--border-color);
+    padding: 15px 0;
+    margin-bottom: 20px;
+    border-radius: 12px 12px 0 0;
+    box-shadow: 0 2px 4px rgba(0,0,0,0.05);
 }
-
-::-webkit-scrollbar-thumb {
-    background: var(--accent-color);
-    border-radius: var(--radius-sm);
+.header-title { 
+    color: var(--primary-color);
+    font-size: 1.8rem;
+    font-weight: 700;
+    text-align: center;
 }
-
-::-webkit-scrollbar-thumb:hover {
-    background: var(--accent-color-alt);
+.header-subtitle { 
+    color: var(--dark-text);
+    font-size: 1rem;
+    text-align: center;
+    margin-top: 5px;
+}
+.chat-container { 
+    border-radius: 12px !important;
+    box-shadow: 0 4px 6px rgba(0,0,0,0.1) !important;
+    background-color: var(--white) !important;
+    border: 1px solid var(--border-color) !important;
+    min-height: 500px;
+}
+.tool-container { 
+    background-color: var(--white);
+    border-radius: 12px;
+    box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+    padding: 15px;
+    margin-bottom: 20px;
+}
+.code-block { 
+    background-color: var(--code-bg);
+    padding: 12px;
+    border-radius: 8px;
+    font-family: 'Roboto Mono', monospace;
+    overflow-x: auto;
+    margin: 10px 0;
+    border-left: 3px solid var(--primary-color);
 }
 """
 
@@ -439,7 +453,7 @@ def process_code_file(file_obj):
         language = detect_language(file_extension)
         
         # Calculate metrics
-        metrics = enhanced_code_analysis(content, language)
+        metrics = calculate_complexity_metrics(content, language)
         
         # Create vectorstore if embeddings are available
         session_id = None
@@ -471,18 +485,86 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     </div>
     """)
     
-    # Main container with all functionality in tabs
-    with gr.Tabs() as main_tabs:
-        # Chat Assistant Tab
-        with gr.TabItem("💬 Chat Assistant", id=0):
-            with gr.Row():
-                with gr.Column(scale=1):
-                    model_dropdown = gr.Dropdown(
-                        choices=["llama3-70b-8192", "mixtral-8x7b-32768", "gemma-7b-it"],
-                        value="llama3-70b-8192",
-                        label="Model Selection"
-                    )
+    with gr.Row(elem_classes="container"):
+        with gr.Column(scale=1, min_width=300):
+            file_input = gr.File(
+                label="Upload Code File",
+                file_types=[".py", ".js", ".java", ".cpp", ".c", ".cs", ".php", ".rb", ".go", ".ts"],
+                type="binary"
+            )
+            upload_button = gr.Button("Analyze Code", variant="primary")
+            file_status = gr.Markdown("No file uploaded yet")
+            model_dropdown = gr.Dropdown(
+                choices=["llama3-70b-8192", "mixtral-8x7b-32768", "gemma-7b-it"],
+                value="llama3-70b-8192",
+                label="Select Model"
+            )
             
+            # Developer Tools Section
+            gr.Markdown("### Developer Tools", elem_classes="tool-title")
+            with gr.Group(elem_classes="tool-container"):  # Replace Box with Group
+                with gr.Tabs():
+                    with gr.TabItem("GitHub Search"):
+                        repo_query = gr.Textbox(label="Search Query", placeholder="Enter keywords to search for repositories")
+                        with gr.Row():
+                            language = gr.Dropdown(
+                                choices=["any", "JavaScript", "Python", "Java", "C++", "TypeScript", "Go", "Rust", "PHP", "C#"],
+                                value="any",
+                                label="Language"
+                            )
+                            min_stars = gr.Dropdown(
+                                choices=["0", "10", "50", "100", "1000", "10000"],
+                                value="0",
+                                label="Min Stars"
+                            )
+                        sort_by = gr.Dropdown(
+                            choices=["stars", "forks", "updated"],
+                            value="stars",
+                            label="Sort By"
+                        )
+                        repo_search_btn = gr.Button("Search Repositories")
+                    
+                    with gr.TabItem("Stack Overflow"):
+                        stack_query = gr.Textbox(label="Search Query", placeholder="Enter your technical question")
+                        with gr.Row():
+                            tag = gr.Dropdown(
+                                choices=["any", "python", "javascript", "java", "c++", "react", "node.js", "android", "ios", "sql"],
+                                value="any",
+                                label="Tag"
+                            )
+                            so_sort_by = gr.Dropdown(
+                                choices=["votes", "newest", "activity"],
+                                value="votes",
+                                label="Sort By"
+                            )
+                        so_search_btn = gr.Button("Search Stack Overflow")
+                
+                    with gr.TabItem("Code Explainer"):
+                        code_input = gr.Textbox(
+                            label="Code to Explain", 
+                            placeholder="Paste your code here...",
+                            lines=10
+                        )
+                        explain_btn = gr.Button("Explain Code")
+                
+        with gr.Column(scale=2, min_width=600):
+            with gr.Tabs():
+                with gr.TabItem("Code Analysis"):
+                    with gr.Column(elem_classes="code-viewer-container"):
+                        code_metrics = gr.Markdown("No code analyzed yet", elem_classes="stats-box")
+                        code_recommendations = gr.Markdown("", elem_classes="recommendations-box")
+                
+                with gr.TabItem("GitHub Results"):
+                    repo_results = gr.Markdown("Search for repositories to see results here")
+                
+                with gr.TabItem("Stack Overflow Results"):
+                    stack_results = gr.Markdown("Search for questions to see results here")
+                    
+                with gr.TabItem("Code Explanation"):
+                    code_explanation = gr.Markdown("Paste your code and click 'Explain Code' to see an explanation here")
+    
+    with gr.Row(elem_classes="container"):
+        with gr.Column(scale=2, min_width=600):
             chatbot = gr.Chatbot(
                 height=500, 
                 show_copy_button=True, 
@@ -497,73 +579,8 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                 )
                 send_btn = gr.Button("Send", scale=1)
             clear_btn = gr.Button("Clear Conversation")
-        
-        # Code Analysis Tab
-        with gr.TabItem("📄 Code Analysis", id=1):
-            with gr.Row():
-                with gr.Column(scale=1):
-                    file_input = gr.File(
-                        label="Upload Code File",
-                        file_types=[".py", ".js", ".java", ".cpp", ".c", ".cs", ".php", ".rb", ".go", ".ts"],
-                        type="binary"
-                    )
-                    upload_button = gr.Button("Analyze Code", variant="primary")
-                    file_status = gr.Markdown("No file uploaded yet")
-                
-                with gr.Column(scale=2):
-                    code_metrics = gr.Markdown("No code analyzed yet", elem_classes="stats-box")
-                    code_recommendations = gr.Markdown("", elem_classes="recommendations-box")
-        
-        # Developer Tools Tab
-        with gr.TabItem("🔍 Developer Tools", id=2):
-            with gr.Tabs():
-                with gr.TabItem("GitHub Search"):
-                    repo_query = gr.Textbox(label="Search Query", placeholder="Enter keywords to search for repositories")
-                    with gr.Row():
-                        language = gr.Dropdown(
-                            choices=["any", "JavaScript", "Python", "Java", "C++", "TypeScript", "Go", "Rust", "PHP", "C#"],
-                            value="any",
-                            label="Language"
-                        )
-                        min_stars = gr.Dropdown(
-                            choices=["0", "10", "50", "100", "1000", "10000"],
-                            value="0",
-                            label="Min Stars"
-                        )
-                    sort_by = gr.Dropdown(
-                        choices=["stars", "forks", "updated"],
-                        value="stars",
-                        label="Sort By"
-                    )
-                    repo_search_btn = gr.Button("Search Repositories")
-                    repo_results = gr.Markdown("Search for repositories to see results here")
-                
-                with gr.TabItem("Stack Overflow"):
-                    stack_query = gr.Textbox(label="Search Query", placeholder="Enter your technical question")
-                    with gr.Row():
-                        tag = gr.Dropdown(
-                            choices=["any", "python", "javascript", "java", "c++", "react", "node.js", "android", "ios", "sql"],
-                            value="any",
-                            label="Tag"
-                        )
-                        so_sort_by = gr.Dropdown(
-                            choices=["votes", "newest", "activity"],
-                            value="votes",
-                            label="Sort By"
-                        )
-                    so_search_btn = gr.Button("Search Stack Overflow")
-                    stack_results = gr.Markdown("Search for questions to see results here")
-                    
-                with gr.TabItem("Code Explainer"):
-                    code_input = gr.Textbox(
-                        label="Code to Explain", 
-                        placeholder="Paste your code here...",
-                        lines=10
-                    )
-                    explain_btn = gr.Button("Explain Code")
-                    code_explanation = gr.Markdown("Paste your code and click 'Explain Code' to see an explanation here")
     
-    # Event Handlers
+    # Update event handlers
     upload_button.click(
         process_code_file,
         inputs=[file_input],
@@ -582,7 +599,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
         inputs=[code_state],
         outputs=[code_metrics]
     ).then(
-        lambda state: generate_enhanced_recommendations(state),
+        lambda state: generate_recommendations(state),
         inputs=[code_state],
         outputs=[code_recommendations]
     )
@@ -626,15 +643,14 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
 
 # Add footer with attribution
 gr.HTML("""
-<div style="text-align: center; margin-top: 20px; padding: 20px; color: #666; font-size: 0.9rem; border-top: 1px solid #eee;">
-    Created by Calvin Allen-Crawford<br>
-    <span style="font-style: italic; font-size: 0.8rem;">Founder of Cosmick Visions</span>
+<div style="text-align: center; margin-top: 20px; padding: 10px; color: #666; font-size: 0.8rem; border-top: 1px solid #eee;">
+    Created by Calvin Allen Crawford
 </div>
 """)
 
-# Launch the app directly
+# Launch the app
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch() 
 
 # Add new helper functions
 def detect_language(extension):
@@ -669,64 +685,20 @@ def calculate_complexity_metrics(content, language):
     
     return metrics
 
-def enhanced_code_analysis(content, language):
-    """Improved code analysis with more metrics"""
-    metrics = calculate_complexity_metrics(content, language)
-    
-    # Add security scanning
-    security_issues = scan_security_issues(content, language)
-    
-    # Add best practices analysis
-    best_practices = analyze_best_practices(content, language)
-    
-    # Add performance suggestions
-    performance_tips = analyze_performance(content, language)
-    
-    return {
-        **metrics,
-        "security_issues": security_issues,
-        "best_practices": best_practices,
-        "performance_tips": performance_tips
-    }
-
-def generate_enhanced_recommendations(metrics):
-    """Enhanced recommendations with more detail"""
+def generate_recommendations(metrics):
+    """Generate code quality recommendations based on metrics"""
     recommendations = []
     
-    # Existing complexity checks
     if metrics.get("cyclomatic_complexity", 0) > 10:
-        recommendations.append({
-            "type": "complexity",
-            "severity": "warning",
-            "message": "��� High cyclomatic complexity detected",
-            "details": "Consider breaking down complex functions into smaller, more manageable pieces.",
-            "examples": ["Extract repeated logic into helper functions", "Use early returns to reduce nesting"]
-        })
+        recommendations.append("🔄 High cyclomatic complexity detected. Consider breaking down complex functions.")
     
-    # Add security recommendations
-    for issue in metrics.get("security_issues", []):
-        recommendations.append({
-            "type": "security",
-            "severity": issue["severity"],
-            "message": f"🔒 {issue['title']}",
-            "details": issue["description"],
-            "fix": issue["recommendation"]
-        })
+    if metrics.get("code_lines", 0) > 300:
+        recommendations.append("📏 File is quite large. Consider splitting it into multiple modules.")
     
-    return format_recommendations(recommendations)
-
-def scan_security_issues(content, language):
-    # This function is called but not defined
-    pass
-
-def analyze_best_practices(content, language):
-    # This function is called but not defined
-    pass
-
-def analyze_performance(content, language):
-    # This function is called but not defined
-    pass
-
-def format_recommendations(recommendations):
-    # This function is called but not defined
-    pass 
+    if metrics.get("functions", 0) > 10:
+        recommendations.append("🔧 Large number of functions. Consider grouping related functions into classes.")
+    
+    if metrics.get("comments", 0) / max(metrics.get("code_lines", 1), 1) < 0.1:
+        recommendations.append("📝 Low comment ratio. Consider adding more documentation.")
+    
+    return "### Recommendations\n\n" + "\n\n".join(recommendations) if recommendations else ""