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
+from langchain_community.embeddings import SentenceTransformerEmbeddings
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 import fitz  # PyMuPDF
 import base64
@@ -25,21 +25,13 @@ client = groq.Client(api_key=os.getenv("GROQ_TECH_API_KEY"))
 
 # Initialize embeddings with error handling
 try:
-    # Force CPU usage for embeddings
-    embeddings = HuggingFaceInstructEmbeddings(
-        model_name="hkunlp/instructor-base",
-        model_kwargs={"device": "cpu"}  # Force CPU usage
+    embeddings = SentenceTransformerEmbeddings(
+        model_name="all-MiniLM-L6-v2",
+        model_kwargs={"device": "cpu"}
     )
 except Exception as e:
-    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
+    print(f"Error loading embeddings: {e}")
+    embeddings = None
 
 # Directory to store FAISS indexes with better naming
 FAISS_INDEX_DIR = "faiss_indexes_tech_cpu"
@@ -49,69 +41,288 @@ if not os.path.exists(FAISS_INDEX_DIR):
 # Dictionary to store user-specific vectorstores
 user_vectorstores = {}
 
-# Custom CSS for Tech theme
+# Update the custom CSS for a more modern chat interface
 custom_css = """
-:root {
-    --primary-color: #4285F4;
-    --secondary-color: #34A853;
-    --accent-color: #EA4335;
-    --light-background: #F8F9FA;
-    --dark-text: #202124;
-    --white: #FFFFFF;
-    --border-color: #DADCE0;
-    --code-bg: #F1F3F4;
-}
-body { 
-    background-color: var(--light-background); 
-    font-family: 'Google Sans', 'Roboto', sans-serif; 
+body {
+    margin: 0;
+    padding: 0;
+    background-color: #f0f2f5;
+    font-family: 'Inter', -apple-system, BlinkMacSystemFont, sans-serif;
 }
-.container { 
-    max-width: 1200px !important; 
-    margin: 0 auto !important; 
-    padding: 10px; 
+
+/* Main app container */
+.main-app {
+    width: 100%;
+    max-width: 1200px;
+    margin: 0 auto;
+    padding: 20px;
 }
-.header { 
-    background-color: var(--white);
-    border-bottom: 1px solid var(--border-color);
-    padding: 15px 0;
+
+/* Header styling */
+.app-header {
+    background: linear-gradient(135deg, #0061ff 0%, #60efff 100%);
+    padding: 30px;
+    border-radius: 16px;
     margin-bottom: 20px;
-    border-radius: 12px 12px 0 0;
-    box-shadow: 0 2px 4px rgba(0,0,0,0.05);
+    box-shadow: 0 4px 20px rgba(0, 0, 0, 0.1);
+    color: white;
+    text-align: center;
 }
-.header-title { 
-    color: var(--primary-color);
-    font-size: 1.8rem;
+
+.app-header h1 {
+    margin: 0;
+    font-size: 2.5rem;
     font-weight: 700;
-    text-align: center;
+    background: linear-gradient(to right, #fff, #e0e0e0);
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
 }
-.header-subtitle { 
-    color: var(--dark-text);
-    font-size: 1rem;
-    text-align: center;
-    margin-top: 5px;
+
+/* 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;
 }
-.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;
+
+.file-upload-area:hover {
+    border-color: #0061ff !important;
+    background: rgba(0, 97, 255, 0.05) !important;
+}
+
+/* Avatar styling */
+.avatar {
+    width: 32px !important;
+    height: 32px !important;
+    border-radius: 50% !important;
+    margin-right: 12px !important;
 }
-.tool-container { 
-    background-color: var(--white);
+
+/* Message timestamp */
+.message-time {
+    font-size: 12px !important;
+    color: rgba(0, 0, 0, 0.4) !important;
+    margin-top: 4px !important;
+}
+
+/* Loading indicator */
+.typing-indicator {
+    display: flex;
+    align-items: center;
+    gap: 4px;
+    padding: 8px 12px;
+    background: #f8f9fa;
     border-radius: 12px;
-    box-shadow: 0 4px 6px rgba(0,0,0,0.1);
-    padding: 15px;
-    margin-bottom: 20px;
+    width: fit-content;
 }
-.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);
+
+.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');
+        });
+    });
+}
+
+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';
+    });
 }
 """
 
@@ -584,181 +795,96 @@ def process_code_file(file_obj):
     except Exception as e:
         return None, f"Error processing file: {str(e)}", {}
 
-# 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"
-            )
+# 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"
+                )
             
-            # 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):
+            # Chat interface
             chatbot = gr.Chatbot(
-                height=500, 
-                show_copy_button=True, 
-                elem_classes="chat-container",
-                type="messages"
+                elem_classes="message-wrap",
+                show_label=False,
+                height=550,
+                avatar_images=["user.png", "bot.png"],
+                bubble_full_width=False
             )
-            with gr.Row():
+            
+            with gr.Row(elem_classes="input-area"):
                 msg = gr.Textbox(
-                    show_label=False, 
-                    placeholder="Ask about your code, type /github to search repos, or /stack to search Stack Overflow...", 
-                    scale=5
+                    show_label=False,
+                    placeholder="Ask me anything about your code...",
+                    elem_classes="input-textbox",
+                    scale=20
                 )
-                send_btn = gr.Button("Send", scale=1)
-            clear_btn = gr.Button("Clear Conversation")
+                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)
     
-    # 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]
+    # 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]
     )
-    
+
     msg.submit(
-        generate_response,
-        inputs=[msg, current_session_id, model_dropdown, chatbot],
+        chat,
+        inputs=[msg, chatbot, current_session_id],
         outputs=[chatbot]
     ).then(lambda: "", None, [msg])
-    
+
     send_btn.click(
-        generate_response,
-        inputs=[msg, current_session_id, model_dropdown, chatbot],
+        chat,
+        inputs=[msg, chatbot, current_session_id],
         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()