SujalAI commited on
Commit
0b9e5e0
·
verified ·
1 Parent(s): 9f78844

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +39 -0
  2. requirements.txt +2 -0
app.py ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import streamlit as st
3
+ from langchain_openai import ChatOpenAI
4
+
5
+ #Add an API key below to successfuly run the project
6
+ os.environ["OPENAI_API_KEY"] = ""
7
+
8
+ #App UI starts here
9
+ st.set_page_config(page_title="Chatbot demo", page_icon=":robot:")
10
+ st.header("You:")
11
+
12
+ def get_input():
13
+ text_input = st.text_input("You:", key=input)
14
+ return text_input
15
+
16
+ if "sessionMessages" not in st.session_state:
17
+ st.session_state.sessionMessages = [
18
+ ("system", "You are a helpful assistant")
19
+ ]
20
+
21
+ def get_answer(user_input):
22
+ llm = ChatOpenAI(
23
+ model="gpt-3.5-turbo-0125",
24
+ temperature=0,
25
+ max_tokens=None,
26
+ timeout=None,
27
+ max_retries=2,
28
+ )
29
+ st.session_state.sessionMessages.append(("human", user_input))
30
+ ai_msg = llm.invoke(st.session_state.sessionMessages)
31
+ st.session_state.sessionMessages.append(("ai", ai_msg.content))
32
+ return ai_msg.content
33
+
34
+ user_input = get_input()
35
+ Submit = st.button("Generate")
36
+
37
+ if Submit:
38
+ st.subheader("Answer:")
39
+ st.write(get_answer(user_input))
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ langchain-openai
2
+ streamlit