Update app.py

app.py

@@ -24,7 +24,7 @@ class GeminiRAGSystem:
         self.chunks = []
         self.dataset_loaded = False
         self.loading_error = None
-        self.gemini_api_key = os.getenv("AIzaSyASrFvE3gFPigihza0JTuALzZmBx0Kc3d0")  # Changed from hardcoded key
+        self.gemini_api_key = os.getenv("GEMINI_API_KEY")
         
         # Initialize embedding model
         try:
@@ -43,11 +43,11 @@ class GeminiRAGSystem:
         """Load dataset in a background thread"""
         def load_task():
             try:
-                # Load dataset directly without progress bar
+                # Load dataset directly
                 dataset = load_dataset(
                     DATASET_NAME,
                     split='train',
-                    download_config={"use_auth_token": False}
+                    download_mode="force_redownload"  # Fixes extraction error
                 )
                 
                 # Process dataset
@@ -96,10 +96,10 @@ class GeminiRAGSystem:
         """Generate response with robust error handling"""
         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 f"⚠️ Dataset loading failed: {self.loading_error}"
+            return "⚠️ Dataset is still loading, please wait..."
         if not self.gemini_api_key:
-            return " Please set your Gemini API key in environment variables"
+            return "🔑 Please set your Gemini API key in environment variables"
         
         context = self.get_relevant_context(query)
         if not context:
@@ -116,7 +116,7 @@ class GeminiRAGSystem:
             response = model.generate_content(prompt)
             return response.text
         except Exception as e:
-            return f" API Error: {str(e)}"
+            return f"⚠️ API Error: {str(e)}"
 
 # Initialize system
 try:
@@ -126,21 +126,35 @@ except Exception as e:
 
 # Create interface
 with gr.Blocks(title="UE Chatbot") as app:
-    gr.Markdown("# UE 24/7 Service")
+    gr.Markdown("# UE 24 Hour Service")
     
     with gr.Row():
-        chatbot = gr.Chatbot(height=500)
+        chatbot = gr.Chatbot(height=500, label="Chat History", 
+                           avatar_images=(None, (None, "https://huggingface.co/spaces/groq/Groq-LLM/resolve/main/groq_logo.png")),
+                           bubble_full_width=False)
     
     with gr.Row():
-        query = gr.Textbox(label="Your question", placeholder="Ask your question...", scale=4)
+        query = gr.Textbox(label="Your question", 
+                         placeholder="Ask your question...", 
+                         scale=4)
         submit_btn = gr.Button("Submit", variant="primary", scale=1)
     
     with gr.Row():
         clear_btn = gr.Button("Clear Chat", variant="secondary")
 
     # Status indicator
-    status = gr.Textbox(label="System Status", visible=False)
+    status = gr.Textbox(label="System Status", 
+                       value="Initializing...",
+                       interactive=False)
+
+    # Update status periodically
+    def update_status():
+        if rag_system.loading_error:
+            return f"Error: {rag_system.loading_error}"
+        return "Ready" if rag_system.dataset_loaded else "Loading dataset..."
     
+    app.load(update_status, None, status, every=1)
+
     # Event handlers
     def respond(message, chat_history):
         try:
@@ -154,17 +168,9 @@ with gr.Blocks(title="UE Chatbot") as app:
     def clear_chat():
         return []
     
-    def get_status():
-        if rag_system.loading_error:
-            return f"Error: {rag_system.loading_error}"
-        return "Ready" if rag_system.dataset_loaded else "Loading dataset..."
-    
     submit_btn.click(respond, [query, chatbot], [query, chatbot])
     query.submit(respond, [query, chatbot], [query, chatbot])
     clear_btn.click(clear_chat, outputs=chatbot)
-    
-    # Periodically check status (hidden from user)
-    app.load(get_status, None, status, every=1)
 
 if __name__ == "__main__":
     app.launch(share=True)