Update app.py

app.py

@@ -1,9 +1,15 @@
 import streamlit as st
 import google.generativeai as genai
+import os
 
-# Configure Gemini API (Replace with your own API key)
-GOOGLE_API_KEY = "AIzaSyB2Nlj3u9DQ4CbpMJkw70j1MvP4xELcSuo"
-genai.configure(api_key=GOOGLE_API_KEY)
+# Securely retrieve API key from environment variables
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+
+# Ensure the API key is set before configuring Gemini
+if GOOGLE_API_KEY:
+    genai.configure(api_key=GOOGLE_API_KEY)
+else:
+    st.error("❌ API Key is missing! Please set it in Hugging Face Secrets.")
 
 # Function to generate a funny story
 def get_funny_story(query, style, length):
@@ -12,9 +18,9 @@ def get_funny_story(query, style, length):
         f"Make it {length} sentences long and style it as {style}. "
         "The story should be humorous and entertaining."
     )
-    
+
     try:
-        model = genai.GenerativeModel("gemini-2.0-pro-exp-02-05")
+        model = genai.GenerativeModel("gemini-pro")
         response = model.generate_content(prompt)
         return response.text
     except Exception as e:
@@ -43,4 +49,3 @@ if st.button("Generate Funny Story"):
         st.write(story)
     else:
         st.warning("Please enter a query for your story.")
-