Update app.py

app.py

@@ -1,60 +1,41 @@
-# 🧠 SoberUp: Drunk-to-Sober Text Translator (Gradio Edition)
-
-## Install Gemini API & Gradio
-```bash
-pip install -q -U google-genai gradio
-```
-
----
-
-## 📦 `app.py`: The Gradio Interface
-```python
 import gradio as gr
 from google import genai
 import os
 
-# Correct Gemini API setup
-client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
+# Get API key from environment variable (set this in 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.")
+
+# Initialize the Gemini client
+client = genai.Client(api_key=API_KEY)
+
+# Load Gemini model
 model = client.models.get("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 UI
+# Gradio Interface
 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 text here..."),
     outputs="text",
     title="🍺 SoberUp Translator",
-    description="Paste your chaotic, drunk ramblings and get a polished, sober version."
+    description="AI-powered translator for decoding drunk messages into sober, coherent statements using Gemini Flash."
 )
 
-demo.launch()
-```
-
----
-
-## 🧪 Example Prompt:
-```text
-oh mannn i just met a guy at tha taco bell drivethru named jason and we’re startin a band called laser lizards 😂🎸🌮
-```
-
-**Output:**
-```text
-I just met a guy named Jason at Taco Bell. We're starting a band called Laser Lizards.
-```
-
----
-
-## 🔐 Security Tips
-- Use `os.getenv("GEMINI_API_KEY")` from Hugging Face secrets panel
-- Optionally add profanity filtering or moderation layer
-
-Want it hosted on Hugging Face or deployed as a mobile app? Just say the word: `> DEPLOY TO CLOUD` or `> MAKE MOBILE VERSION`.
+# Launch app
+if __name__ == "__main__":
+    demo.launch()