Update app.py

app.py

@@ -50,18 +50,18 @@ def ingest_docs_to_chroma():
     chunks = text_splitter.split_documents(all_docs)
     st.write(f"Total chunks created: {len(chunks)}")
 
+    # Clear existing vectors to avoid duplication
+    collection.delete(where={})
+
     # Add Chunks to ChromaDB
     for chunk in chunks:
-        # Flatten list content if necessary
         if isinstance(chunk.page_content, list):
             content = " ".join(str(item) for item in chunk.page_content).strip()
         else:
             content = str(chunk.page_content).strip()
         
-        metadata = chunk.metadata
         doc_id = str(hash(content))
-        collection.add(documents=[content], metadatas=[metadata], ids=[doc_id])
-
+        collection.add(documents=[content], ids=[doc_id])
 
     st.success("✅ Knowledge Base Updated Successfully!")
 
@@ -81,11 +81,15 @@ def ask_groq(context, question):
         {"role": "system", "content": "You are a helpful assistant. Always provide relevant video and website links if possible."},
         {"role": "user", "content": f"Context:\n{context}\n\nQuestion: {question}\nAnswer (include links):"}
     ]
-    response = groq_client.chat.completions.create(
-        model=GROQ_MODEL,
-        messages=messages
-    )
-    return response.choices[0].message.content.strip()
+    try:
+        response = groq_client.chat.completions.create(
+            model=GROQ_MODEL,
+            messages=messages
+        )
+        return response.choices[0].message.content.strip()
+    except Exception as e:
+        st.error(f"Groq API error: {e}")
+        return "⚠️ Failed to get response from Groq API."
 
 # --- Streamlit UI ---
 def main():
@@ -93,10 +97,8 @@ def main():
     st.title("🎓 EduBot for @icodeguru0")
     st.markdown("Ask anything based on pre-loaded iCodeGuru knowledge.")
 
-    # --- Auto Update Knowledge Base at App Start ---
-    st.info("🔄 Updating Knowledge Base from /docs/...")
-    ingest_docs_to_chroma()
-    st.success("✅ Knowledge Base Loaded Successfully!")
+    if st.button("🔄 Refresh Knowledge Base"):
+        ingest_docs_to_chroma()
 
     st.markdown("---")
 
@@ -105,7 +107,7 @@ def main():
     if user_question:
         vector_context = search_vector_data(user_question)
         if vector_context:
-            with st.spinner("🧠 Answering from knowledge base..."):
+            with st.spinner("🧐 Answering from knowledge base..."):
                 answer = ask_groq(vector_context, user_question)
                 st.success(answer)
         else:
@@ -115,4 +117,4 @@ def main():
     st.caption("Powered by ChromaDB 🧠 and Groq ⚡")
 
 if __name__ == "__main__":
-    main()
+    main()