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"
@@ -802,6 +794,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                     with gr.Row(elem_classes="code-analysis"):
                         with gr.Column(scale=1, elem_classes="analysis-card"):
                             gr.Markdown("### Code Metrics", elem_classes="card-title")
+                            code_metrics = gr.Markdown("Upload code to see metrics", elem_id="code-metrics")
                             with gr.Row(elem_classes="metric-grid"):
                                 gr.Markdown("**Language:**", elem_classes="metric-label")
                                 gr.Markdown("-", elem_classes="metric-value", elem_id="language-metric")
@@ -833,8 +826,8 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                 chatbot = gr.Chatbot(
                     height=400,
                     elem_classes="chat-messages",
-                    bubble_full_width=False,
-                    show_copy_button=True
+                    show_copy_button=True,
+                    type="messages"
                 )
                 
                 with gr.Row(elem_classes="chat-input-area"):