Rozeeeee's picture
Update app.py
3029342 verified
Raw
History Blame Contribute Delete
1.58 kB
import streamlit as st
import google.generativeai as genai
# Configure Gemini API
genai.configure(api_key='AIzaSyBDvXBds9i0TFZL6c7d7KACti6P1M7U_gc')
model = genai.GenerativeModel('gemini-pro')
def generate_response(user_message):
try:
response = model.generate_content(user_message)
if response.text:
return response.text
else:
return "Error: The model generated an empty response."
except Exception as e:
return f"An error occurred: {str(e)}"
# Initialize session state for chat history
if 'chat_history' not in st.session_state:
st.session_state.chat_history = []
st.title("Gemini QA System")
st.write("Ask a question and get an answer from Gemini AI.")
# Text input for user question
user_message = st.text_area("Enter your question here...", height=100)
# Button to submit the question
if st.button("Get Response"):
# Generate response
response = generate_response(user_message)
# Add to chat history
st.session_state.chat_history.append(("You", user_message))
st.session_state.chat_history.append(("Gemini", response))
# Display chat history
st.write("### Chat History")
for role, message in st.session_state.chat_history:
st.text_area(f"{role}:", value=message, height=100, disabled=True)
# Examples section
st.write("### Examples")
st.write("- What is the capital of France?")
st.write("- Explain quantum computing in simple terms.")
# Clear chat history button
if st.button("Clear Chat History"):
st.session_state.chat_history = []
st.experimental_rerun()