Upload app.py

app.py

@@ -29,10 +29,13 @@ from groq import Groq
 # ══════════════════════════════════════════════════════════
 CHROMA_PATH    = "./chroma_ena_db"
 
-if os.path.exists("chroma_ena_db.zip") and not os.path.exists(CHROMA_PATH):
+if os.path.exists("chroma_ena_db.zip"):
     try:
+        import shutil
+        # Always overwrite if there's a fresh zip
         with zipfile.ZipFile("chroma_ena_db.zip", 'r') as zip_ref:
             zip_ref.extractall(".")
+        os.remove("chroma_ena_db.zip")  # Delete zip after extraction so we don't do it again
     except Exception as e:
         print(f"Zip extraction failed: {e}")
 
@@ -151,13 +154,14 @@ def load_llm():
     return Groq(api_key=key) if key else None
 
 @st.cache_resource
-def load_all_chunks(_vdb, _force_reload_v2=True):
+def load_all_chunks(_vdb, _force_reload_v3=True):
     try:
         col = _vdb._collection.get(include=["documents", "metadatas"])
         docs  = col.get("documents") or []
         metas = col.get("metadatas") or []
         return docs, metas
-    except Exception:
+    except Exception as e:
+        print(f"Exception in load_all_chunks: {e}")
         return [], []
 
 # ══════════════════════════════════════════════════════════
@@ -670,7 +674,9 @@ with st.sidebar:
     # DB status — آمن هنا خارج cache_resource
     chunks_count = len(load_all_chunks(vectordb)[0])
     if chunks_count == 0:
-        st.error("❌ DB فارغة — شغّل build_chroma.py")
+        st.error("❌ DB فارغة — أو جاري التحديث (اضغط مسح المحادثة للتحديث)")
+        load_vectordb.clear()
+        load_all_chunks.clear()
     else:
         st.success(f"✅ DB: {chunks_count} chunks")
     st.metric("💬 عدد الأسئلة", len([m for m in st.session_state.chat_messages if m["role"]=="user"]))