Update app.py

app.py

@@ -2,17 +2,14 @@ import gradio as gr
 import requests
 import os
 
-# Load Gemini API key securely from Hugging Face Secrets
+# Securely get the API key from Hugging Face secrets
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 
 def query_gemini(prompt):
-    url = "https://generativelanguage.googleapis.com/v1/models/gemini-1.5-flash-latest:generateContent"
+    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent?key={GEMINI_API_KEY}"
     headers = {
         "Content-Type": "application/json"
     }
-    params = {
-        "key": GEMINI_API_KEY
-    }
     data = {
         "contents": [
             {
@@ -23,19 +20,19 @@ def query_gemini(prompt):
         ]
     }
 
-    response = requests.post(url, headers=headers, params=params, json=data)
+    response = requests.post(url, headers=headers, json=data)
 
     try:
         return response.json()['candidates'][0]['content']['parts'][0]['text']
     except Exception as e:
-        return f"Error: {e}\nResponse: {response.text}"
+        return f"Error: {e}\nFull Response: {response.text}"
 
 iface = gr.Interface(
     fn=query_gemini,
-    inputs=gr.Textbox(lines=4, placeholder="Ask a question to Gemini..."),
+    inputs=gr.Textbox(lines=4, placeholder="Ask something..."),
     outputs="text",
-    title="Gemini 1.5 Flash API with Gradio",
-    description="Interact with Gemini 1.5 Flash through Hugging Face Space"
+    title="Gemini 2.0 Flash with Gradio",
+    description="Powered by Google's Gemini 2.0 Flash API"
 )
 
 if __name__ == "__main__":