Update app.py

app.py

@@ -6,8 +6,10 @@ import faiss
 from sentence_transformers import SentenceTransformer
 from datasets import load_dataset
 from dotenv import load_dotenv
-import asyncio
-import time
+import warnings
+
+# Suppress warnings
+warnings.filterwarnings("ignore")
 
 # Load environment variables
 load_dotenv()
@@ -19,6 +21,10 @@ DATASET_NAME = "midrees2806/7K_Dataset"
 CHUNK_SIZE = 500
 TOP_K = 3
 
+# Workaround for huggingface_hub compatibility
+import huggingface_hub
+huggingface_hub.__version__ = "0.13.4"
+
 class GeminiRAGSystem:
     def __init__(self):
         self.index = None
@@ -37,11 +43,11 @@ class GeminiRAGSystem:
         if self.gemini_api_key:
             genai.configure(api_key=self.gemini_api_key)
         
-        # Start dataset loading
+        # Load dataset
         self.load_dataset()
     
     def load_dataset(self):
-        """Load dataset synchronously"""
+        """Load dataset synchronously with error handling"""
         try:
             # Load dataset directly
             dataset = load_dataset(
@@ -94,9 +100,9 @@ class GeminiRAGSystem:
         if not self.dataset_loaded:
             if self.loading_error:
                 return f"⚠️ Dataset loading failed: {self.loading_error}"
-            return "⚠️ Dataset is still loading, please wait..."
+            return "⚠️ System initialization in progress..."
         if not self.gemini_api_key:
-            return "🔑 Please set your Gemini API key in environment variables"
+            return "🔑 API key not configured"
         
         context = self.get_relevant_context(query)
         if not context:
@@ -128,7 +134,6 @@ with gr.Blocks(title="UE Chatbot") as app:
     with gr.Row():
         chatbot = gr.Chatbot(
             height=500,
-            avatar_images=(None, (None, "https://huggingface.co/spaces/groq/Groq-LLM/resolve/main/groq_logo.png")),
             bubble_full_width=False
         )
     
@@ -146,7 +151,7 @@ with gr.Blocks(title="UE Chatbot") as app:
     # Status indicator
     status = gr.Textbox(
         label="System Status",
-        value="Loading dataset..." if not rag_system.dataset_loaded else "Ready",
+        value="Ready" if rag_system.dataset_loaded else f"Initializing... {rag_system.loading_error or ''}",
         interactive=False
     )