Update app.py

app.py

@@ -1,32 +1,29 @@
 """
-Hugging Face Spaces version of the chatbot
-Provides both Gradio interface and API endpoints
+Hugging Face Spaces version of the chatbot - Gradio only
+For API integration, use the original Flask app.py
 """
 import gradio as gr
-from flask import Flask, request, jsonify
-from flask_cors import CORS
-from dotenv import load_dotenv
 import os
 import requests
 from vector_store import query_vector_store, init_vector_store
 import logging
-import threading
 
 logging.basicConfig(level=logging.INFO)
-load_dotenv()
 
+# Get API key from environment (HF Spaces secrets)
 DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
 
-# Initialize Flask app for API
-app = Flask(__name__)
-CORS(app)
-
 # Initialize vector store
+print("Initializing vector store...")
 init_vector_store()
+print("Vector store initialized!")
 
 def chat_with_eve(message, history):
     """Gradio chat function"""
     try:
+        if not message.strip():
+            return "Please enter a message!"
+        
         # Get context from vector store
         context = query_vector_store(message)
         
@@ -40,7 +37,10 @@ def chat_with_eve(message, history):
         payload = {
             "model": "deepseek/deepseek-chat-v3-0324",
             "messages": [
-                {"role": "system", "content": f"You are a museum assistant known as Eve with a warm personality. Use this context:\n{context}"},
+                {
+                    "role": "system", 
+                    "content": f"You are Eve, a warm and knowledgeable museum assistant. You help visitors learn about art, history, culture, and museum collections. Use this context to provide helpful, engaging responses: {context}"
+                },
                 {"role": "user", "content": message}
             ]
         }
@@ -51,88 +51,48 @@ def chat_with_eve(message, history):
             json=payload,
             timeout=30
         )
-        response.raise_for_status()
-        data = response.json()
-
-        if "choices" in data and len(data["choices"]) > 0:
-            return data["choices"][0]["message"]["content"]
+        
+        if response.status_code == 200:
+            data = response.json()
+            if "choices" in data and len(data["choices"]) > 0:
+                return data["choices"][0]["message"]["content"]
+            else:
+                return "I'm sorry, I didn't get a proper response. Please try again!"
         else:
-            return "I'm sorry, I couldn't process your request right now."
+            return f"I'm having trouble connecting right now (Error: {response.status_code}). Please try again in a moment!"
             
+    except requests.exceptions.Timeout:
+        return "The request timed out. Please try again with a shorter message."
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Request error: {str(e)}")
+        return "I'm having connection issues. Please try again in a moment."
     except Exception as e:
         logging.error(f"Chat error: {str(e)}")
-        return "I'm having trouble connecting right now. Please try again."
-
-# Flask API endpoints
-@app.route("/chat", methods=["POST"])
-def chat_api():
-    """API endpoint for external services"""
-    try:
-        user_input = request.json.get("message")
-        if not user_input:
-            return jsonify({"error": "No message provided"}), 400
-
-        context = query_vector_store(user_input)
-
-        headers = {
-            "Authorization": f"Bearer {DEEPSEEK_API_KEY}",
-            "Content-Type": "application/json",
-            "HTTP-Referer": "https://huggingface.co/spaces/",
-            "X-Title": "mAIseums ChatBot"
-        }
+        return "Something went wrong. Please try again or rephrase your question."
 
-        payload = {
-            "model": "deepseek/deepseek-chat-v3-0324",
-            "messages": [
-                {"role": "system", "content": f"You are a museum assistant known as Eve with a warm personality. Use this context:\n{context}"},
-                {"role": "user", "content": user_input}
-            ]
-        }
-
-        response = requests.post(
-            "https://openrouter.ai/api/v1/chat/completions",
-            headers=headers,
-            json=payload,
-            timeout=30
-        )
-        response.raise_for_status()
-        data = response.json()
-
-        if "choices" in data and len(data["choices"]) > 0:
-            return jsonify({"response": data["choices"][0]["message"]["content"]})
-        else:
-            return jsonify({"error": "No response from AI service"}), 500
-
-    except Exception as e:
-        logging.error(f"API error: {str(e)}")
-        return jsonify({"error": "Internal server error", "details": str(e)}), 500
-
-@app.route("/health", methods=["GET"])
-def health():
-    return jsonify({"status": "healthy", "service": "mAIseums-chatbot"}), 200
-
-def run_flask():
-    """Run Flask in a separate thread"""
-    app.run(host="0.0.0.0", port=7860, debug=False)
-
-# Create Gradio interface
+# Create the Gradio interface
 demo = gr.ChatInterface(
-    chat_with_eve,
+    fn=chat_with_eve,
     title="🏛️ mAIseums - Meet Eve, Your Museum Assistant",
-    description="Ask me anything about museums, art, history, and cultural heritage!",
-    theme="soft",
+    description="""
+    Welcome to mAIseums! I'm Eve, your virtual museum guide. Ask me anything about:
+    • Museum collections and exhibits
+    • Art history and cultural heritage  
+    • Historical artifacts and their stories
+    • Visiting information and museum tips
+    
+    I'm here to make your museum experience educational and enjoyable! ✨
+    """,
     examples=[
         "Tell me about the museum's art collection",
-        "What exhibitions do you have?",
-        "Can you explain this artifact?",
-        "What are your visiting hours?",
+        "What can you tell me about ancient civilizations?",
+        "I'm interested in Renaissance paintings",
+        "What exhibitions do you currently have?",
+        "Can you explain the history behind this artifact?",
+        "What are your visiting hours and ticket prices?"
     ]
 )
 
 if __name__ == "__main__":
-    # Start Flask API in background thread for external API access
-    flask_thread = threading.Thread(target=run_flask, daemon=True)
-    flask_thread.start()
-    
-    # Launch Gradio interface (this will be the main HF Spaces interface)
-    demo.launch()
+    # Launch the Gradio app
+    demo.launch()