PK commited on
Commit
803885a
·
verified ·
1 Parent(s): 3e1f824

Update streamlit_app.py

Browse files
Files changed (1) hide show
  1. streamlit_app.py +7 -11
streamlit_app.py CHANGED
@@ -59,19 +59,15 @@ def _remove_all_videos():
59
 
60
  expander = st.sidebar.expander("Options", expanded=True)
61
  # ✅ Configure Gemini API
62
- API_KEYi = expander.text_input("Google API Key", "AIzaSyB2uDj6IeuNszSuk80feW-eBgpIDH0WFSY")
63
-
64
- # ✅ Load API key securely
65
- API_KEYe = os.getenv("GOOGLE_API_KEY")
66
-
67
  # ✅ Check if API key is available
68
- if not API_KEYi or API_KEYe:
69
  st.error("⚠️ API key is missing! Set GOOGLE_API_KEY as an environment variable")
70
- st.stop()
71
- if API_KEYi:
72
- genai.configure(api_key=API_KEYi)
73
- if API_KEYe:
74
- genai.configure(api_key=API_KEYe)
75
 
76
  #Safety Settings
77
  safety_settings = [ { "category": "HARM_CATEGORY_HARASSMENT", "threshold": "OFF" }, { "category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "OFF" }, { "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "OFF" }, { "category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "OFF" }, ]
 
59
 
60
  expander = st.sidebar.expander("Options", expanded=True)
61
  # ✅ Configure Gemini API
62
+ API_KEY = expander.text_input("Google API Key", "AIzaSyB2uDj6IeuNszSuk80feW-eBgpIDH0WFSY")
 
 
 
 
63
  # ✅ Check if API key is available
64
+ if not API_KEY:
65
  st.error("⚠️ API key is missing! Set GOOGLE_API_KEY as an environment variable")
66
+ # ✅ Load API key securely
67
+ API_KEY = os.getenv("GOOGLE_API_KEY")
68
+
69
+ if API_KEY:
70
+ genai.configure(api_key=API_KEY)
71
 
72
  #Safety Settings
73
  safety_settings = [ { "category": "HARM_CATEGORY_HARASSMENT", "threshold": "OFF" }, { "category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "OFF" }, { "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "OFF" }, { "category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "OFF" }, ]