Update app.py

app.py

@@ -2,40 +2,38 @@ import gradio as gr
 from google import genai
 import os
 
-# Get API key from environment variable (set this in Hugging Face Secrets)
+# Get API key from Hugging Face secrets
 API_KEY = os.getenv("GEMINI_API_KEY")
 
 if not API_KEY:
-    raise EnvironmentError("Missing GEMINI_API_KEY. Set it in Hugging Face Spaces → Secrets.")
+    raise EnvironmentError("Missing GEMINI_API_KEY. Make sure it's set in the Hugging Face Secrets.")
 
-# Initialize the Gemini client
+# Initialize Gemini Client
 client = genai.Client(api_key=API_KEY)
 
-# Load Gemini model
-model = client.models.get("models/gemini-2.0-flash")
+# Use correct keyword argument for model name
+model = client.models.get(name="models/gemini-2.0-flash")
 
-# 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)
         return response.text
     except Exception as e:
         return f"Error: {e}"
 
-# Gradio Interface
+# Gradio interface
 demo = gr.Interface(
     fn=sober_up,
-    inputs=gr.Textbox(lines=4, placeholder="Paste your chaotic, drunk text here..."),
+    inputs=gr.Textbox(lines=4, placeholder="Type or paste your drunk text here..."),
     outputs="text",
     title="🍺 SoberUp Translator",
-    description="AI-powered translator for decoding drunk messages into sober, coherent statements using Gemini Flash."
+    description="AI-powered Gemini app that translates your drunk ramblings into clean, sober text."
 )
 
-# Launch app
 if __name__ == "__main__":
     demo.launch()