import streamlit as st
import google.generativeai as genai

# Directly configure API key (Replace with your actual API key)
genai.configure(api_key="AIzaSyD5UTVMIDrIQQwl9vRJl1aS-mEEz3gv06M")  # <-- Use string directly

# Function to generate a funny short story
def generate_funny_story(query):
    model = genai.GenerativeModel("gemini-2.0-pro-exp-02-05")
    prompt = f"Write a short and funny story based on this idea: {query}. Make it hilarious!"
    
    try:
        response = model.generate_content(prompt)
        return response.text
    except Exception as e:
        return f"Error: {str(e)}"

# Streamlit UI
st.title("😂 AI Funny Story Generator")
st.write("Give me an idea, and I'll turn it into a short, funny story!")

# User input
user_query = st.text_input("Enter your story idea (e.g., 'A dog that becomes a detective')")

# Generate story button
if st.button("Generate Story"):
    if user_query:
        story = generate_funny_story(user_query)
        st.subheader("🤣 Here’s Your Funny Story:")
        st.write(story)
    else:
        st.warning("Please enter an idea first!")

# Footer
st.markdown("---")
st.caption("Powered by Google Gemini 🚀")