admin08077 commited on
Commit
c2d69a4
·
verified ·
1 Parent(s): 0bbc10a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -9
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 Hugging Face Secrets
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
- # Proper Gemini initialization
12
  client = genai.Client(api_key=API_KEY)
13
 
14
- # Correct way to load the Gemini 2.0 Flash model
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 UI
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)