Update app.py

app.py

@@ -8,15 +8,15 @@ from langchain_text_splitters import RecursiveCharacterTextSplitter
 from langchain_huggingface import HuggingFaceEmbeddings
 from langchain_community.vectorstores import FAISS
 
-# Suppress technical warnings for a clean logs
+# Suppress technical warnings
 warnings.filterwarnings("ignore")
 
 # --- 1. CONFIGURATION & SECRETS ---
-# On Hugging Face, set 'MY_GROQ_SECRET' in Settings > Variables and Secrets
+# Ensure 'MY_GROQ_SECRET' is added in Hugging Face Settings > Secrets
 GROQ_API_KEY = os.environ.get("MY_GROQ_SECRET")
 client = Groq(api_key=GROQ_API_KEY)
 
-# YOUR HIDDEN LINKS (Never shown in UI)
+# HIDDEN DATA SOURCE
 GDRIVE_LINKS = [
     "https://drive.google.com/file/d/10D3uJqBYG9gMWsNHcpTW4I6BKmA2otfH/view?usp=sharing"
 ]
@@ -35,7 +35,6 @@ def download_gdrive_pdf(url, output_path):
         return False
     return False
 
-# Initialize the vector database on startup
 all_chunks = []
 text_splitter = RecursiveCharacterTextSplitter(chunk_size=800, chunk_overlap=150)
 
@@ -52,18 +51,17 @@ for i, link in enumerate(GDRIVE_LINKS):
 embeddings = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
 vector_db = FAISS.from_documents(all_chunks, embeddings)
 
-# --- 3. STRICT RAG LOGIC ---
+# --- 3. RAG LOGIC ---
 def respond(message, history):
-    # Retrieve relevant snippets
+    # Find context snippets
     docs = vector_db.similarity_search(message, k=5)
     context = "\n\n".join([doc.page_content for doc in docs])
     
-    # Strict instructions: No outside knowledge allowed
     system_prompt = f"""
     You are a professional Knowledge Assistant. 
-    1. Answer ONLY using the context provided.
+    1. Answer ONLY using the context provided below.
     2. If the answer is NOT in the context, say: "Answer not found in provided documents."
-    3. Do not mention the context or the technical nature of the search.
+    3. Keep answers direct and factual.
     
     CONTEXT:
     {context}
@@ -79,31 +77,30 @@ def respond(message, history):
     )
     return chat_completion.choices[0].message.content
 
-# --- 4. ATTRACTIVE MODERN UI ---
+# --- 4. MODERN UI DESIGN ---
 custom_css = """
 body { background-color: #0f172a; }
-.gradio-container { max-width: 850px !important; margin: auto; padding-top: 50px; }
-#title-text { text-align: center; color: #38bdf8; font-weight: 800; margin-bottom: 5px; }
-#desc-text { text-align: center; color: #94a3b8; margin-bottom: 25px; }
-.chat-container { border-radius: 20px !important; border: 1px solid #334155 !important; background: #1e293b !important; box-shadow: 0 20px 50px rgba(0,0,0,0.4); }
-.primary-btn { background: linear-gradient(135deg, #38bdf8, #818cf8) !important; border: none !important; color: white !important; }
+.gradio-container { max-width: 850px !important; margin: auto; padding-top: 30px; }
+#title-text { text-align: center; color: #38bdf8; font-weight: 800; }
+#desc-text { text-align: center; color: #94a3b8; margin-bottom: 20px; }
+.chat-container { border-radius: 15px !important; border: 1px solid #334155 !important; background: #1e293b !important; }
 footer { display: none !important; }
 """
 
-with gr.Blocks(theme=gr.themes.Default(primary_hue="sky"), css=custom_css) as demo:
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="sky"), css=custom_css) as demo:
     gr.HTML("<h1 id='title-text'>🌀 DocuVortex</h1>")
-    # REPLACE "User's Research AI" with your own name here!
-    gr.HTML("<p id='desc-text'>User's Research AI: Strict Document Knowledge Base</p>")
+    gr.HTML("<p id='desc-text'>Bilal's Research AI: Strict Document Knowledge Base</p>")
     
     with gr.Column(elem_id="chat-container"):
         gr.ChatInterface(
             fn=respond,
-            chatbot=gr.Chatbot(height=550, bubble_full_width=False, show_label=False),
-            textbox=gr.Textbox(placeholder="Ask a question about the document...", container=False, scale=7),
-            submit_btn=gr.Button("Ask AI", variant="primary", elem_classes="primary-btn"),
+            # FIXED: Removed 'bubble_full_width' which caused the crash
+            chatbot=gr.Chatbot(height=550, show_label=False),
+            textbox=gr.Textbox(placeholder="Ask Bilal's AI a question...", container=False, scale=7),
+            submit_btn=gr.Button("Ask AI", variant="primary"),
             retry_btn=None,
             undo_btn=None,
-            clear_btn=gr.Button("New Chat", variant="secondary")
+            clear_btn=gr.Button("Refresh Chat", variant="secondary")
         )
 
 if __name__ == "__main__":