shaun10 commited on
Commit
105aa5f
·
verified ·
1 Parent(s): e9791d0

Upload simple_chatbot.py

Browse files
Files changed (1) hide show
  1. simple_chatbot.py +31 -0
simple_chatbot.py ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import os
3
+ os.getenv("GOOGLE_API_KEY")
4
+ from langchain_google_genai import ChatGoogleGenerativeAI
5
+
6
+
7
+ def llm_response(question):
8
+ llm = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.5)
9
+ response = llm.invoke(question)
10
+ return response.content
11
+
12
+
13
+ st.sidebar.title("LangChain QA Chatbot")
14
+ query = st.sidebar.text_area("Ask Anything....", height=250)
15
+ generate_button = st.sidebar.button("Generate Answer", use_container_width=True)
16
+ response = llm_response(query)
17
+
18
+ if generate_button:
19
+ if len(query) == 0:
20
+ st.header("Please Ask Your Question.")
21
+ else:
22
+ st.header("Generated Content")
23
+ st.write(response)
24
+
25
+
26
+ # streamlit run simple_chatbot.py
27
+
28
+
29
+
30
+
31
+