kcdw commited on
Commit
fc99cc4
·
verified ·
1 Parent(s): 87743b9

create streamlit_app.py

Browse files
Files changed (1) hide show
  1. streamlit_app.py +27 -0
streamlit_app.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import google.generativeai as genai
3
+ import os
4
+
5
+ # Load API key from environment variable
6
+ genai.configure(api_key=os.getenv("GOOGLE_GENAI_API_KEY"))
7
+
8
+ # Title and description
9
+ st.title("Gemini AI Chatbot")
10
+ st.write("Chatbot powered by Google's Gemini API")
11
+
12
+ # User input
13
+ user_prompt = st.text_area("Enter your question:", height=150)
14
+
15
+ if st.button("Generate Response"):
16
+ if user_prompt.strip():
17
+ try:
18
+ with st.spinner("Fetching response..."):
19
+ response = genai.GenerativeModel("gemini-1.5-flash").generate_content(
20
+ "Assume I am Sahan. " + user_prompt
21
+ )
22
+ st.success("Response:")
23
+ st.write(response.text)
24
+ except Exception as e:
25
+ st.error(f"An error occurred: {str(e)}")
26
+ else:
27
+ st.warning("Please enter a valid prompt.")