Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 import requests
+import os
 
-# Replace this with your actual Gemini 2.0 API key
-GEMINI_API_KEY = "AIzaSyDcLzXpiN9Wzen6jNL3suHenSKUeGniUbc"
+# Retrieve the API key from the environment variable
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 
 def query_gemini(prompt):
-    url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent?key=GEMINI_API_KEY"
+    url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent"
     headers = {
         "Content-Type": "application/json"
     }