Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,19 +2,16 @@ import gradio as gr
|
|
| 2 |
from google import genai
|
| 3 |
import os
|
| 4 |
|
| 5 |
-
# Get API key from
|
| 6 |
API_KEY = os.getenv("GEMINI_API_KEY")
|
| 7 |
|
| 8 |
if not API_KEY:
|
| 9 |
raise EnvironmentError("GEMINI_API_KEY is not set. Add it under 'Secrets' in Hugging Face Spaces.")
|
| 10 |
|
| 11 |
-
#
|
| 12 |
client = genai.Client(api_key=API_KEY)
|
| 13 |
|
| 14 |
-
#
|
| 15 |
-
model = client.models.generate_content
|
| 16 |
-
|
| 17 |
-
# Define translation logic
|
| 18 |
def sober_up(drunk_text):
|
| 19 |
if not drunk_text.strip():
|
| 20 |
return "Please enter a message."
|
|
@@ -30,14 +27,15 @@ def sober_up(drunk_text):
|
|
| 30 |
except Exception as e:
|
| 31 |
return f"Error: {e}"
|
| 32 |
|
| 33 |
-
# Gradio
|
| 34 |
demo = gr.Interface(
|
| 35 |
fn=sober_up,
|
| 36 |
inputs=gr.Textbox(lines=4, placeholder="Paste your chaotic, drunk message here..."),
|
| 37 |
outputs="text",
|
| 38 |
title="🍺 SoberUp Translator",
|
| 39 |
-
description="Turn drunk ramblings into sober brilliance using Gemini 2.0 Flash."
|
|
|
|
| 40 |
)
|
| 41 |
|
| 42 |
if __name__ == "__main__":
|
| 43 |
-
demo.launch()
|
|
|
|
| 2 |
from google import genai
|
| 3 |
import os
|
| 4 |
|
| 5 |
+
# Get API key from environment
|
| 6 |
API_KEY = os.getenv("GEMINI_API_KEY")
|
| 7 |
|
| 8 |
if not API_KEY:
|
| 9 |
raise EnvironmentError("GEMINI_API_KEY is not set. Add it under 'Secrets' in Hugging Face Spaces.")
|
| 10 |
|
| 11 |
+
# Initialize Gemini client
|
| 12 |
client = genai.Client(api_key=API_KEY)
|
| 13 |
|
| 14 |
+
# Define the translation logic
|
|
|
|
|
|
|
|
|
|
| 15 |
def sober_up(drunk_text):
|
| 16 |
if not drunk_text.strip():
|
| 17 |
return "Please enter a message."
|
|
|
|
| 27 |
except Exception as e:
|
| 28 |
return f"Error: {e}"
|
| 29 |
|
| 30 |
+
# Define the Gradio interface
|
| 31 |
demo = gr.Interface(
|
| 32 |
fn=sober_up,
|
| 33 |
inputs=gr.Textbox(lines=4, placeholder="Paste your chaotic, drunk message here..."),
|
| 34 |
outputs="text",
|
| 35 |
title="🍺 SoberUp Translator",
|
| 36 |
+
description="Turn drunk ramblings into sober brilliance using Gemini 2.0 Flash.\n\n🚨 Instead of sharing: just COPY THE TEXT MESSAGE to save your drunk ass. Share link's just a myth, like your dignity last night.",
|
| 37 |
+
allow_flagging="never"
|
| 38 |
)
|
| 39 |
|
| 40 |
if __name__ == "__main__":
|
| 41 |
+
demo.launch(share=False)
|