import streamlit as st
st.set_page_config(page_title="Hugging Face Space", page_icon=":rocket:")
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("Gemini AI x DataSciencePro")

logo_image = "Wallpaper_Company.png"
st.image(logo_image, use_column_width=True)

# 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)