Update app.py

app.py

@@ -2,19 +2,19 @@ import gradio as gr
 from google import genai
 import os
 
-# Load API key from environment variable
+# Get API key from Hugging Face Secrets
 API_KEY = os.getenv("GEMINI_API_KEY")
 
 if not API_KEY:
-    raise EnvironmentError("GEMINI_API_KEY is not set. Add it as a secret in Hugging Face Spaces.")
+    raise EnvironmentError("GEMINI_API_KEY is not set. Add it under 'Secrets' in Hugging Face Spaces.")
 
-# Configure Gemini client
-genai.configure(api_key=API_KEY)
+# ✅ Proper Gemini initialization
+client = genai.Client(api_key=API_KEY)
 
-# Load model correctly using get_model
-model = genai.GenerativeModel("gemini-2.0-flash")
+# ✅ Correct way to load the Gemini 2.0 Flash model
+model = client.models.generate_content
 
-# Define the translation function
+# Define translation logic
 def sober_up(drunk_text):
     if not drunk_text.strip():
         return "Please enter a message."
@@ -22,18 +22,21 @@ def sober_up(drunk_text):
     prompt = f"Translate this drunk message into a sober, coherent version: '{drunk_text}'"
     
     try:
-        response = model.generate_content(prompt)
+        response = client.models.generate_content(
+            model="gemini-2.0-flash",
+            contents=prompt
+        )
         return response.text
     except Exception as e:
         return f"Error: {e}"
 
-# Launch Gradio UI
+# Gradio UI
 demo = gr.Interface(
     fn=sober_up,
     inputs=gr.Textbox(lines=4, placeholder="Paste your chaotic, drunk message here..."),
     outputs="text",
     title="🍺 SoberUp Translator",
-    description="Turn drunk messages into sober clarity with Gemini 2.0 Flash."
+    description="Turn drunk ramblings into sober brilliance using Gemini 2.0 Flash."
 )
 
 if __name__ == "__main__":