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,288 +49,69 @@ if not os.path.exists(FAISS_INDEX_DIR):
 # Dictionary to store user-specific vectorstores
 user_vectorstores = {}
 
-# Update the custom CSS for a more modern chat interface
+# Custom CSS for Tech theme
 custom_css = """
-body {
-    margin: 0;
-    padding: 0;
-    background-color: #f0f2f5;
-    font-family: 'Inter', -apple-system, BlinkMacSystemFont, sans-serif;
+:root {
+    --primary-color: #4285F4;
+    --secondary-color: #34A853;
+    --accent-color: #EA4335;
+    --light-background: #F8F9FA;
+    --dark-text: #202124;
+    --white: #FFFFFF;
+    --border-color: #DADCE0;
+    --code-bg: #F1F3F4;
 }
-
-/* Main app container */
-.main-app {
-    width: 100%;
-    max-width: 1200px;
-    margin: 0 auto;
-    padding: 20px;
+body { 
+    background-color: var(--light-background); 
+    font-family: 'Google Sans', 'Roboto', sans-serif; 
 }
-
-/* Header styling */
-.app-header {
-    background: linear-gradient(135deg, #0061ff 0%, #60efff 100%);
-    padding: 30px;
-    border-radius: 16px;
+.container { 
+    max-width: 1200px !important; 
+    margin: 0 auto !important; 
+    padding: 10px; 
+}
+.header { 
+    background-color: var(--white);
+    border-bottom: 1px solid var(--border-color);
+    padding: 15px 0;
     margin-bottom: 20px;
-    box-shadow: 0 4px 20px rgba(0, 0, 0, 0.1);
-    color: white;
-    text-align: center;
+    border-radius: 12px 12px 0 0;
+    box-shadow: 0 2px 4px rgba(0,0,0,0.05);
 }
-
-.app-header h1 {
-    margin: 0;
-    font-size: 2.5rem;
+.header-title { 
+    color: var(--primary-color);
+    font-size: 1.8rem;
     font-weight: 700;
-    background: linear-gradient(to right, #fff, #e0e0e0);
-    -webkit-background-clip: text;
-    -webkit-text-fill-color: transparent;
-}
-
-/* Chat container styling */
-.chat-container {
-    background-color: #ffffff;
-    border-radius: 16px;
-    box-shadow: 0 4px 20px rgba(0, 0, 0, 0.08);
-    overflow: hidden;
-    height: 700px !important;
-    display: flex !important;
-    flex-direction: column !important;
-    transition: all 0.3s ease;
-}
-
-/* Message container */
-.message-wrap {
-    flex-grow: 1 !important;
-    overflow-y: auto !important;
-    padding: 24px !important;
-    scroll-behavior: smooth;
-}
-
-.message-wrap::-webkit-scrollbar {
-    width: 6px;
-}
-
-.message-wrap::-webkit-scrollbar-track {
-    background: #f1f1f1;
-    border-radius: 3px;
-}
-
-.message-wrap::-webkit-scrollbar-thumb {
-    background: #c1c1c1;
-    border-radius: 3px;
-}
-
-/* Message bubbles */
-.message {
-    max-width: 85% !important;
-    margin: 12px 0 !important;
-    padding: 16px 20px !important;
-    line-height: 1.5 !important;
-    font-size: 15px !important;
-    transition: all 0.2s ease;
-}
-
-.user-message {
-    background: linear-gradient(135deg, #0061ff 0%, #60a5fa 100%) !important;
-    color: white !important;
-    margin-left: auto !important;
-    border-radius: 18px 18px 0 18px !important;
-    box-shadow: 0 2px 8px rgba(0, 97, 255, 0.15) !important;
-}
-
-.bot-message {
-    background: #f8f9fa !important;
-    color: #1a1a1a !important;
-    margin-right: auto !important;
-    border-radius: 18px 18px 18px 0 !important;
-    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.05) !important;
-    border: 1px solid #e8e8e8 !important;
-}
-
-/* Code blocks in messages */
-.bot-message code {
-    background: #2d3748 !important;
-    color: #e2e8f0 !important;
-    padding: 12px 16px !important;
-    border-radius: 8px !important;
-    font-family: 'Fira Code', monospace !important;
-    margin: 8px 0 !important;
-    display: block;
-    white-space: pre-wrap;
-}
-
-/* Input area */
-.input-area {
-    padding: 24px !important;
-    background: #ffffff !important;
-    border-top: 1px solid rgba(0, 0, 0, 0.06) !important;
-    display: flex !important;
-    align-items: center !important;
-    gap: 12px !important;
-}
-
-.input-textbox {
-    background: #f8f9fa !important;
-    border: 1px solid #e8e8e8 !important;
-    border-radius: 24px !important;
-    padding: 16px 24px !important;
-    font-size: 15px !important;
-    transition: all 0.2s ease !important;
-    box-shadow: 0 2px 6px rgba(0, 0, 0, 0.02) !important;
-}
-
-.input-textbox:focus {
-    border-color: #0061ff !important;
-    box-shadow: 0 2px 12px rgba(0, 97, 255, 0.1) !important;
-    background: white !important;
-}
-
-.send-button {
-    background: linear-gradient(135deg, #0061ff 0%, #60a5fa 100%) !important;
-    color: white !important;
-    border-radius: 50% !important;
-    width: 46px !important;
-    height: 46px !important;
-    padding: 0 !important;
-    display: flex !important;
-    align-items: center !important;
-    justify-content: center !important;
-    cursor: pointer !important;
-    transition: all 0.2s ease !important;
-    box-shadow: 0 2px 8px rgba(0, 97, 255, 0.2) !important;
-}
-
-.send-button:hover {
-    transform: translateY(-1px) !important;
-    box-shadow: 0 4px 12px rgba(0, 97, 255, 0.3) !important;
-}
-
-/* File upload area */
-.file-upload-area {
-    text-align: center !important;
-    padding: 40px !important;
-    border: 2px dashed #c1c1c1 !important;
-    border-radius: 16px !important;
-    margin: 24px !important;
-    transition: all 0.3s ease !important;
-    background: rgba(0, 97, 255, 0.02) !important;
-}
-
-.file-upload-area:hover {
-    border-color: #0061ff !important;
-    background: rgba(0, 97, 255, 0.05) !important;
+    text-align: center;
 }
-
-/* Avatar styling */
-.avatar {
-    width: 32px !important;
-    height: 32px !important;
-    border-radius: 50% !important;
-    margin-right: 12px !important;
+.header-subtitle { 
+    color: var(--dark-text);
+    font-size: 1rem;
+    text-align: center;
+    margin-top: 5px;
 }
-
-/* Message timestamp */
-.message-time {
-    font-size: 12px !important;
-    color: rgba(0, 0, 0, 0.4) !important;
-    margin-top: 4px !important;
+.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;
 }
-
-/* Loading indicator */
-.typing-indicator {
-    display: flex;
-    align-items: center;
-    gap: 4px;
-    padding: 8px 12px;
-    background: #f8f9fa;
+.tool-container { 
+    background-color: var(--white);
     border-radius: 12px;
-    width: fit-content;
-}
-
-.typing-dot {
-    width: 6px;
-    height: 6px;
-    background: #0061ff;
-    border-radius: 50%;
-    animation: typing 1.4s infinite;
-}
-
-@keyframes typing {
-    0%, 100% { transform: translateY(0); }
-    50% { transform: translateY(-4px); }
-}
-"""
-
-# 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();
-}
-
-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';
-        }
-    });
-}
-
-function setupTabs() {
-    const tabs = document.querySelectorAll('.tab');
-    tabs.forEach(tab => {
-        tab.addEventListener('click', () => {
-            tabs.forEach(t => t.classList.remove('active'));
-            tab.classList.add('active');
-        });
-    });
+    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);
 }
 """
 
@@ -795,96 +584,181 @@ def process_code_file(file_obj):
     except Exception as e:
         return None, f"Error processing file: {str(e)}", {}
 
-# Update the Gradio interface layout
-with gr.Blocks(css=custom_css) as demo:
-    with gr.Column(elem_classes="main-app"):
-    # Header
-        with gr.Column(elem_classes="app-header"):
-            gr.Markdown("# Tech-Vision AI")
-            gr.Markdown("Your AI-powered code analysis assistant")
-        
-        # Chat container
-        with gr.Column(elem_classes="chat-container"):
-            # File upload view
-            with gr.Column(visible=True) as upload_view:
-                file_upload = gr.File(
-                    label="Drop your code file here or click to upload",
-                    file_types=[".txt", ".py", ".js", ".java", ".cpp", ".html", ".css"],
-                    elem_classes="file-upload-area"
-                )
+# 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("""
+    <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"):
+        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"
+            )
             
-            # Chat interface
+            # 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(
-                elem_classes="message-wrap",
-                show_label=False,
-                height=550,
-                avatar_images=["user.png", "bot.png"],
-                bubble_full_width=False
+                height=500, 
+                show_copy_button=True, 
+                elem_classes="chat-container",
+                type="messages"
             )
-            
-            with gr.Row(elem_classes="input-area"):
+            with gr.Row():
                 msg = gr.Textbox(
-                    show_label=False,
-                    placeholder="Ask me anything about your code...",
-                    elem_classes="input-textbox",
-                    scale=20
+                    show_label=False, 
+                    placeholder="Ask about your code, type /github to search repos, or /stack to search Stack Overflow...", 
+                    scale=5
                 )
-                send_btn = gr.Button("→", elem_classes="send-button", scale=1)
-
-    # State for tracking if file is uploaded
-    file_uploaded = gr.State(False)
-    current_session_id = gr.State(None)
+                send_btn = gr.Button("Send", scale=1)
+            clear_btn = gr.Button("Clear Conversation")
     
-    # Handle file upload
-    def process_file_and_start_chat(file_obj):
-        if file_obj is None:
-            return None, gr.update(visible=True), gr.update(value="Please upload a file to begin.")
-        
-        try:
-            session_id, summary = process_code_file(file_obj)
-            
-            # Generate initial message about the uploaded file
-            initial_message = f"I've analyzed your uploaded file. Here's what I found:\n\n{summary}\n\nWhat would you like to know about it?"
-            
-            return (
-                session_id,
-                gr.update(visible=False),  # Hide upload view
-                [[None, initial_message]]  # Add initial bot message
-            )
-        except Exception as e:
-            return None, gr.update(visible=True), gr.update(value=f"Error processing file: {str(e)}")
-
-    # Update chat handling
-    def chat(message, history, session_id):
-        if not message:
-            return history
-        
-        try:
-            # Use the existing generate_response function with the session context
-            updated_history = generate_response(message, session_id, "llama3-70b-8192", history)
-            return updated_history
-        except Exception as e:
-            history.append([message, f"Error: {str(e)}"])
-            return history
-
-    # Set up event handlers
-    file_upload.upload(
-        process_file_and_start_chat,
-        inputs=[file_upload],
-        outputs=[current_session_id, upload_view, chatbot]
+    # 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(
-        chat,
-        inputs=[msg, chatbot, current_session_id],
+        generate_response,
+        inputs=[msg, current_session_id, model_dropdown, chatbot],
         outputs=[chatbot]
     ).then(lambda: "", None, [msg])
-
+    
     send_btn.click(
-        chat,
-        inputs=[msg, chatbot, current_session_id],
+        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 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()