Update app.py

app.py

@@ -12,23 +12,32 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-# Use the key directly from the environment
-genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+# Read Gemini API key from Hugging Face secret
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+if not GEMINI_API_KEY:
+    raise RuntimeError("GEMINI_API_KEY not found in environment.")
 
-# Optional debug
-print("Gemini key loaded:", os.getenv("GEMINI_API_KEY") is not None)
-
-model = genai.GenerativeModel("gemini-2.5-flash")
+def call_gemini(prompt: str) -> str:
+    """Use Gemini via REST API instead of gRPC-based SDK"""
+    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={GEMINI_API_KEY}"
+    payload = {
+        "contents": [{"parts": [{"text": prompt}]}]
+    }
+    response = requests.post(url, json=payload)
+    try:
+        return response.json()["candidates"][0]["content"]["parts"][0]["text"]
+    except Exception:
+        raise HTTPException(status_code=500, detail="Error in Gemini response format.")
 
 app = FastAPI()
 
-# Add CORS middleware
+# CORS
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Allows all origins
+    allow_origins=["*"],
     allow_credentials=True,
-    allow_methods=["*"],  # Allows all methods
-    allow_headers=["*"],  # Allows all headers
+    allow_methods=["*"],
+    allow_headers=["*"],
 )
 
 # Initialize enhanced components