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 SentenceTransformerEmbeddings
+from langchain_community.embeddings import HuggingFaceInstructEmbeddings
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 import fitz  # PyMuPDF
 import base64
@@ -25,13 +25,21 @@ client = groq.Client(api_key=os.getenv("GROQ_TECH_API_KEY"))
 
 # Initialize embeddings with error handling
 try:
-    embeddings = SentenceTransformerEmbeddings(
-        model_name="all-MiniLM-L6-v2",
-        model_kwargs={"device": "cpu"}
+    # Force CPU usage for embeddings
+    embeddings = HuggingFaceInstructEmbeddings(
+        model_name="hkunlp/instructor-base",
+        model_kwargs={"device": "cpu"}  # Force CPU usage
     )
 except Exception as e:
-    print(f"Error loading embeddings: {e}")
-    embeddings = None
+    print(f"Warning: Failed to load primary embeddings model: {e}")
+    try:
+        embeddings = HuggingFaceInstructEmbeddings(
+            model_name="all-MiniLM-L6-v2",
+            model_kwargs={"device": "cpu"}  # Force CPU usage
+        )
+    except Exception as e:
+        print(f"Warning: Failed to load fallback embeddings model: {e}")
+        embeddings = None
 
 # Directory to store FAISS indexes with better naming
 FAISS_INDEX_DIR = "faiss_indexes_tech_cpu"
@@ -41,246 +49,69 @@ if not os.path.exists(FAISS_INDEX_DIR):
 # Dictionary to store user-specific vectorstores
 user_vectorstores = {}
 
-# Custom CSS to match HTML exactly
+# Custom CSS for Tech theme
 custom_css = """
 :root {
     --primary-color: #4285F4;
     --secondary-color: #34A853;
     --accent-color: #EA4335;
-    --yellow-color: #FBBC05;
     --light-background: #F8F9FA;
     --dark-text: #202124;
     --white: #FFFFFF;
     --border-color: #DADCE0;
     --code-bg: #F1F3F4;
-    --shadow-sm: 0 1px 3px rgba(0,0,0,0.12), 0 1px 2px rgba(0,0,0,0.24);
-    --shadow-md: 0 4px 6px rgba(0,0,0,0.1);
-    --shadow-lg: 0 10px 20px rgba(0,0,0,0.1);
-    --transition: all 0.3s cubic-bezier(.25,.8,.25,1);
-}
-
-/* Global Styles */
-body {
-    background-color: var(--light-background);
-    font-family: 'Roboto', sans-serif;
-    color: var(--dark-text);
-    line-height: 1.6;
-}
-
-/* Header */
-.header {
-    background-color: var(--white);
-    box-shadow: var(--shadow-sm);
-    position: sticky;
-    top: 0;
-    z-index: 100;
-    padding: 16px 0;
-}
-
-.logo {
-    display: flex;
-    align-items: center;
-    gap: 12px;
-}
-
-.logo-icon {
-    color: var(--primary-color);
-    font-size: 24px;
 }
-
-/* Main Layout */
-.container {
-    max-width: 1400px;
-    margin: 0 auto;
-    padding: 40px 20px;
+body { 
+    background-color: var(--light-background); 
+    font-family: 'Google Sans', 'Roboto', sans-serif; 
 }
-
-.main-content {
-    display: grid;
-    grid-template-columns: 320px 1fr;
-    gap: 30px;
+.container { 
+    max-width: 1200px !important; 
+    margin: 0 auto !important; 
+    padding: 10px; 
 }
-
-/* Sidebar */
-.sidebar {
+.header { 
     background-color: var(--white);
-    border-radius: 12px;
-    box-shadow: var(--shadow-md);
-    overflow: hidden;
-}
-
-.sidebar-section {
-    padding: 20px;
     border-bottom: 1px solid var(--border-color);
-}
-
-/* File Upload */
-.file-upload {
-    border: 2px dashed var(--border-color);
-    border-radius: 8px;
-    padding: 20px;
-    text-align: center;
-    cursor: pointer;
-    transition: var(--transition);
-    margin-bottom: 16px;
-}
-
-/* Tabs */
-.tabs {
-    display: flex;
-    background-color: var(--white);
+    padding: 15px 0;
+    margin-bottom: 20px;
     border-radius: 12px 12px 0 0;
-    overflow: hidden;
-    box-shadow: var(--shadow-sm);
+    box-shadow: 0 2px 4px rgba(0,0,0,0.05);
 }
-
-.tab {
-    padding: 16px 24px;
-    font-family: 'Google Sans', sans-serif;
-    font-weight: 500;
-    cursor: pointer;
-    transition: var(--transition);
-    border-bottom: 3px solid transparent;
-}
-
-.tab.active {
+.header-title { 
     color: var(--primary-color);
-    border-bottom-color: var(--primary-color);
-}
-
-/* Chat Section */
-.chat-section {
-    background-color: var(--white);
-    border-radius: 12px;
-    box-shadow: var(--shadow-md);
-    overflow: hidden;
-    margin-top: 24px;
-}
-
-.chat-header {
-    padding: 16px;
-    background-color: var(--primary-color);
-    color: var(--white);
-    display: flex;
-    justify-content: space-between;
-    align-items: center;
-}
-
-.chat-messages {
-    height: 400px;
-    overflow-y: auto;
-    padding: 16px;
-}
-
-.message {
-    max-width: 80%;
-    padding: 12px 16px;
-    border-radius: 18px;
-    margin-bottom: 12px;
-}
-
-.message-user {
-    background-color: var(--primary-color);
-    color: var(--white);
-    margin-left: auto;
-    border-radius: 18px 18px 4px 18px;
+    font-size: 1.8rem;
+    font-weight: 700;
+    text-align: center;
 }
-
-.message-bot {
-    background-color: var(--light-background);
+.header-subtitle { 
     color: var(--dark-text);
-    margin-right: auto;
-    border-radius: 18px 18px 18px 4px;
-}
-
-/* Buttons */
-.primary-button {
-    background-color: var(--primary-color);
-    color: var(--white);
-    border: none;
-    border-radius: 4px;
-    padding: 10px 16px;
-    font-family: 'Google Sans', sans-serif;
-    font-weight: 500;
-    cursor: pointer;
-    transition: var(--transition);
-}
-
-.primary-button:hover {
-    background-color: #3367d6;
-}
-"""
-
-# Custom JavaScript to enhance UI
-custom_js = """
-document.addEventListener('DOMContentLoaded', function() {
-    // Add Font Awesome
-    const fontAwesome = document.createElement('link');
-    fontAwesome.rel = 'stylesheet';
-    fontAwesome.href = 'https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css';
-    document.head.appendChild(fontAwesome);
-    
-    // Add Google Fonts
-    const googleFonts = document.createElement('link');
-    googleFonts.rel = 'stylesheet';
-    googleFonts.href = 'https://fonts.googleapis.com/css2?family=Google+Sans:wght@400;500;700&family=Roboto:wght@300;400;500&family=Roboto+Mono&display=swap';
-    document.head.appendChild(googleFonts);
-    
-    // Initialize UI enhancements
-    setTimeout(enhanceUI, 1000);
-});
-
-function enhanceUI() {
-    // Add icons to headers
-    addIconToHeader('Upload Code', 'fa-upload');
-    addIconToHeader('Developer Tools', 'fa-tools');
-    addIconToHeader('Tech Assistant', 'fa-robot');
-    
-    // Setup tabs
-    setupTabs();
-    
-    // Setup file upload area
-    setupFileUpload();
+    font-size: 1rem;
+    text-align: center;
+    margin-top: 5px;
 }
-
-function addIconToHeader(text, iconClass) {
-    document.querySelectorAll('h3').forEach(header => {
-        if (header.textContent.includes(text)) {
-            const icon = document.createElement('i');
-            icon.className = `fas ${iconClass}`;
-            header.insertBefore(icon, header.firstChild);
-            header.style.display = 'flex';
-            header.style.alignItems = 'center';
-            header.style.gap = '8px';
-        }
-    });
+.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;
 }
-
-function setupTabs() {
-    const tabs = document.querySelectorAll('.tab');
-    tabs.forEach(tab => {
-        tab.addEventListener('click', () => {
-            tabs.forEach(t => t.classList.remove('active'));
-            tab.classList.add('active');
-        });
-    });
+.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;
 }
-
-function setupFileUpload() {
-    const dropzone = document.querySelector('.file-upload');
-    if (!dropzone) return;
-    
-    dropzone.addEventListener('dragover', (e) => {
-        e.preventDefault();
-        dropzone.style.borderColor = 'var(--primary-color)';
-        dropzone.style.backgroundColor = 'rgba(66, 133, 244, 0.05)';
-    });
-    
-    dropzone.addEventListener('dragleave', (e) => {
-        e.preventDefault();
-        dropzone.style.borderColor = 'var(--border-color)';
-        dropzone.style.backgroundColor = 'transparent';
-    });
+.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);
 }
 """
 
@@ -753,106 +584,181 @@ def process_code_file(file_obj):
     except Exception as e:
         return None, f"Error processing file: {str(e)}", {}
 
-# Create the Gradio interface
-with gr.Blocks(css=custom_css, js=custom_js, theme=gr.themes.Soft()) as demo:
-    # Header
+# Update the Gradio interface
+with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
+    current_session_id = gr.State(None)
+    code_state = gr.State({})
+    
     gr.HTML("""
-    <header class="header">
-        <div class="container">
-            <div class="logo">
-                <i class="fas fa-code logo-icon"></i>
-                <span class="logo-text">Tech-Vision AI</span>
-            </div>
-        </div>
-    </header>
+    <div class="header">
+        <div class="header-title">Tech-Vision AI</div>
+        <div class="header-subtitle">Advanced Code Analysis & Technical Assistant</div>
+    </div>
     """)
     
-    with gr.Row(elem_classes="container main-content"):
-        # Sidebar
-        with gr.Column(scale=1, min_width=320, elem_classes="sidebar"):
-            with gr.Group(elem_classes="sidebar-section"):
-                gr.Markdown("### Upload Code")
-                file_input = gr.File(
-                    label="Drag & drop your code file here",
-                    file_types=[".py", ".js", ".java", ".cpp", ".c", ".cs", ".php", ".rb", ".go", ".ts"],
-                    elem_classes="file-upload"
-                )
-                analyze_btn = gr.Button("Analyze Code", elem_classes="primary-button")
-                
-                model_dropdown = gr.Dropdown(
-                    choices=["llama3-70b-8192", "mixtral-8x7b-32768", "gemma-7b-it"],
-                    value="llama3-70b-8192",
-                    label="Select Model"
-                )
+    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"
+            )
             
-            with gr.Group(elem_classes="sidebar-section"):
-                gr.Markdown("### Developer Tools")
+            # 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")
+                        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++"],
+                                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"],
+                                choices=["0", "10", "50", "100", "1000", "10000"],
                                 value="0",
                                 label="Min Stars"
                             )
-                        repo_search_btn = gr.Button("Search", elem_classes="primary-button")
+                        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")
-                        tag = gr.Dropdown(
-                            choices=["any", "python", "javascript", "java"],
-                            value="any",
-                            label="Tag"
+                        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
                         )
-                        stack_search_btn = gr.Button("Search", elem_classes="primary-button")
-        
-        # Main Area
-        with gr.Column(scale=3):
-            with gr.Tabs(elem_classes="tabs"):
+                        explain_btn = gr.Button("Explain Code")
+                
+        with gr.Column(scale=2, min_width=600):
+            with gr.Tabs():
                 with gr.TabItem("Code Analysis"):
-                    with gr.Row():
-                        with gr.Column():
-                            gr.Markdown("### Code Metrics")
-                            metrics = gr.JSON(label="")
-                        with gr.Column():
-                            gr.Markdown("### Recommendations")
-                            recommendations = gr.Markdown()
+                    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()
+                    repo_results = gr.Markdown("Search for repositories to see results here")
                 
-                with gr.TabItem("Stack Results"):
-                    stack_results = gr.Markdown()
-            
-            # Chat Section
-            with gr.Group(elem_classes="chat-section"):
-                with gr.Row(elem_classes="chat-header"):
-                    gr.Markdown("### Tech Assistant")
-                    clear_btn = gr.Button("Clear", elem_classes="chat-control-btn")
-                
-                chatbot = gr.Chatbot(
-                    height=400,
-                    elem_classes="chat-messages",
-                    show_copy_button=True,
-                    type="messages"
+                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, 
+                elem_classes="chat-container",
+                type="messages"
+            )
+            with gr.Row():
+                msg = gr.Textbox(
+                    show_label=False, 
+                    placeholder="Ask about your code, type /github to search repos, or /stack to search Stack Overflow...", 
+                    scale=5
                 )
-                
-                with gr.Row():
-                    msg = gr.Textbox(
-                        placeholder="Ask about your code...",
-                        show_label=False,
-                        elem_classes="chat-input"
-                    )
-                    send_btn = gr.Button("Send", elem_classes="primary-button")
+                send_btn = gr.Button("Send", scale=1)
+            clear_btn = gr.Button("Clear Conversation")
+    
+    # Update event handlers
+    upload_button.click(
+        lambda x: process_code_file(x),
+        inputs=[file_input],
+        outputs=[current_session_id, file_status, code_state]
+    ).then(
+        lambda state: (
+            f"### Code Analysis Results\n\n"
+            f"**Language:** {state.get('language', 'Unknown')}\n"
+            f"**Total Lines:** {state.get('total_lines', 0)}\n"
+            f"**Code Lines:** {state.get('code_lines', 0)}\n"
+            f"**Comment Lines:** {state.get('comments', 0)}\n"
+            f"**Functions:** {state.get('functions', 0)}\n"
+            f"**Classes:** {state.get('classes', 0)}\n"
+            f"**Complexity Score:** {state.get('cyclomatic_complexity', 0)}\n"
+        ),
+        inputs=[code_state],
+        outputs=[code_metrics]
+    ).then(
+        lambda state: generate_recommendations(state),
+        inputs=[code_state],
+        outputs=[code_recommendations]
+    )
+    
+    msg.submit(
+        generate_response,
+        inputs=[msg, current_session_id, model_dropdown, chatbot],
+        outputs=[chatbot]
+    ).then(lambda: "", None, [msg])
+    
+    send_btn.click(
+        generate_response,
+        inputs=[msg, current_session_id, model_dropdown, chatbot],
+        outputs=[chatbot]
+    ).then(lambda: "", None, [msg])
+    
+    clear_btn.click(
+        lambda: ([], None, "No file uploaded", {}, None),
+        None,
+        [chatbot, current_session_id, file_status, code_state, code_metrics]
+    )
+    
+    # Tech tool handlers
+    repo_search_btn.click(
+        perform_repo_search,
+        inputs=[repo_query, language, sort_by, min_stars],
+        outputs=[repo_results]
+    )
+    
+    so_search_btn.click(
+        perform_stack_search,
+        inputs=[stack_query, tag, so_sort_by],
+        outputs=[stack_results]
+    )
+    
+    explain_btn.click(
+        explain_code,
+        inputs=[code_input],
+        outputs=[code_explanation]
+    )
 
-    # Add event handlers (implementation details in previous messages)
-    # ... (event handlers remain the same)
+# Add footer with attribution
+gr.HTML("""
+<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
 if __name__ == "__main__":
     demo.launch()