from dotenv import load_dotenv
import streamlit as st
import os
import google.generativeai as genai

load_dotenv()

# Configure Generative AI model with API key
genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
model = genai.GenerativeModel('gemini-pro')

def get_response(question):
    chat = model.start_chat(history=[])
    response = chat.send_message(question, stream=True)
    return response

st.header('Gemini Chat App')

# Use st.text_area for multiline input
input_text = st.text_area("Input", key='input', height=100)

# Use st.form to encapsulate form elements
with st.form(key='chat_form'):
    submit = st.form_submit_button('Ask')

if submit:
    if input_text.strip():
        response = get_response(input_text)
        st.header('Answer')
        for chunk in response:
            st.write(chunk.text)
        response.resolve()  # Allow response to complete
    else:
        st.warning("Please enter a question.")