Update app.py

app.py

@@ -2,37 +2,38 @@ import gradio as gr
 from google import genai
 import os
 
-# Get API key from Hugging Face secrets
+# Load API key from environment variable
 API_KEY = os.getenv("GEMINI_API_KEY")
 
 if not API_KEY:
-    raise EnvironmentError("Missing GEMINI_API_KEY. Make sure it's set in the Hugging Face Secrets.")
+    raise EnvironmentError("GEMINI_API_KEY is not set. Add it as a secret in Hugging Face Spaces.")
 
-# Initialize Gemini Client
-client = genai.Client(api_key=API_KEY)
+# Configure Gemini client
+genai.configure(api_key=API_KEY)
 
-# Use correct keyword argument for model name
-model = client.models.get(name="models/gemini-2.0-flash")
+# Load model correctly using get_model
+model = genai.GenerativeModel("gemini-2.0-flash")
 
+# Define the translation function
 def sober_up(drunk_text):
     if not drunk_text.strip():
         return "Please enter a message."
     
     prompt = f"Translate this drunk message into a sober, coherent version: '{drunk_text}'"
-
+    
     try:
-        response = model.generate_content(contents=prompt)
+        response = model.generate_content(prompt)
         return response.text
     except Exception as e:
         return f"Error: {e}"
 
-# Gradio interface
+# Launch Gradio UI
 demo = gr.Interface(
     fn=sober_up,
-    inputs=gr.Textbox(lines=4, placeholder="Type or paste your drunk text here..."),
+    inputs=gr.Textbox(lines=4, placeholder="Paste your chaotic, drunk message here..."),
     outputs="text",
     title="🍺 SoberUp Translator",
-    description="AI-powered Gemini app that translates your drunk ramblings into clean, sober text."
+    description="Turn drunk messages into sober clarity with Gemini 2.0 Flash."
 )
 
 if __name__ == "__main__":