import streamlit as st
import os
import google.generativeai as genai
GOOGLE_API_KEY=os.getenv('GEMINI_API_KEY')
genai.configure(api_key=GOOGLE_API_KEY)
model = genai.GenerativeModel(model_name = "gemini-pro")


# Create a banner using Markdown
st.markdown(
    """
    <div style="min-width: 1000px;">
        <div style="background-color:#f63366;padding:10px;border-radius:10px;">
            <h1 style="color:white;text-align:center;">Red Octopus</h1>
        </div>
        <div style="color:black;text-align:center;">
            <h1>Welcome to the Proposition Management Tool</h1>
        </div>
    </div>
    """,
    unsafe_allow_html=True
)

selectedCity = st.selectbox("Please select the City and the Bank Product for Your Proposition.", ["CharlesTown", "Limburg"])
selectedProduct = st.selectbox("Please select the Product", ["Current", "Mortage", "Credit Card", "Crypto"])
userProposal = st.text_input("Enter your Proposition for Select City and Product")

submit_button = st.button("Submit")

if submit_button:
    st.write("You clicked the Submit button!")
    st.write("Entered text:", userProposal)
    response = model.generate_content([userProposal])
    st.write(response.text)