import streamlit as st

import google.generativeai as genai

# Configure Google GenAI
api_key = 'AIzaSyA7jXPscK6XqztKHkZB-QDE_OjpEzZQDCU'
genai.configure(api_key=api_key)

# Initialize GenerativeModel
model = genai.GenerativeModel('gemini-pro')

# Streamlit app
st.title("Hugging Face Space with Streamlit")

# Input text box for user query
user_query = st.text_input("Enter your query:")

if st.button("Generate"):
    # Generate content using the model
    response = model.generate_content(user_query)

    # Display the generated content using Markdown
    st.markdown(response.text)

# Run the app
if __name__ == "__main__":
    st.set_page_config(page_title="Hugging Face Space", page_icon=":rocket:")
    st.run_app()