Update app.py

app.py

@@ -10,12 +10,26 @@ from datetime import datetime
 # 🔑 CONFIG
 # =========================
 
+# Configure Gemini API Key
 os.environ["GOOGLE_API_KEY"] = "AIzaSyBzSyySlN8ELAY3cCZp3tzWTC3YHua1Ej0"
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 model = genai.GenerativeModel("gemini-2.5-flash")
 
-# Initialize Firebase
-cred = credentials.Certificate("firebase_key.json")
+# =========================
+# 🔥 FIREBASE SETUP (with secret)
+# =========================
+
+# Try loading Firebase key from Hugging Face Secrets first
+firebase_key_content = os.getenv("FIREBASE_KEY")
+
+if firebase_key_content:
+    print("✅ Using Firebase key from secret environment variable")
+    cred_dict = json.loads(firebase_key_content)
+    cred = credentials.Certificate(cred_dict)
+else:
+    print("⚠️ Using local firebase_key.json (for local dev only)")
+    cred = credentials.Certificate("firebase_key.json")
+
 firebase_admin.initialize_app(cred)
 db = firestore.client()
 
@@ -23,9 +37,11 @@ db = firestore.client()
 # ⚡ FASTAPI SETUP
 # =========================
 
-app = FastAPI(title="🌾 Smart Farmer Advisor (Safe Memory)",
-              description="Gemini AI with persistent memory stored in user subcollection (safe for frontend)",
-              version="3.1")
+app = FastAPI(
+    title="🌾 Smart Farmer Advisor (Safe Memory)",
+    description="Gemini AI with persistent memory stored in user subcollection (safe for frontend)",
+    version="3.1"
+)
 
 app.add_middleware(
     CORSMiddleware,
@@ -47,6 +63,7 @@ def get_user_chat_history(uid: str, session_id: str = "session_001"):
         return doc.to_dict().get("history", [])
     return []
 
+
 def save_user_chat_history(uid: str, history: list, session_id: str = "session_001"):
     """Save chat history under user's subcollection."""
     chat_doc = db.collection("users").document(uid).collection("chatbot_history").document(session_id)
@@ -116,6 +133,7 @@ def get_schemes(
     data = get_farmer_schemes(query, uid, session_id)
     return {"uid": uid, "session_id": session_id, "query": query, "schemes": data}
 
+
 @app.get("/reset_chat")
 def reset_chat(uid: str = Query(...), session_id: str = Query("session_001")):
     db.collection("users").document(uid).collection("chatbot_history").document(session_id).delete()
@@ -127,5 +145,5 @@ def reset_chat(uid: str = Query(...), session_id: str = Query("session_001")):
 
 if __name__ == "__main__":
     import uvicorn
-    print("🚜 Smart Farmer Advisor (Safe Memory) running at http://127.0.0.1:8005")
-    uvicorn.run(app, host="0.0.0.0", port=8005)
+    print("🚜 Smart Farmer Advisor (Safe Memory) running at http://127.0.0.1:7860")
+    uvicorn.run(app, host="0.0.0.0", port=7860)