garyd1 commited on
Commit
289b1cb
·
verified ·
1 Parent(s): 304ffc8

Beta-version

Browse files
Files changed (1) hide show
  1. app.py +30 -0
app.py CHANGED
@@ -16,9 +16,39 @@ st.header("Welcome, How can I help you?")
16
  #Configure the key
17
  os.environ["GOOGLE_API_KEY"] = keyfile.GOOGLEKEY
18
 
 
 
 
 
 
19
 
20
  #User message
21
  def get_text():
22
  input_text=st.text_input("You: ", key=input)
23
  return input_text
24
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
  #Configure the key
17
  os.environ["GOOGLE_API_KEY"] = keyfile.GOOGLEKEY
18
 
19
+ #General Instruction
20
+ if "sessionMessages" not in st.session_state:
21
+ st.session_state.sessionMessage=[
22
+ SystemMessage(content="You are a medievel magical healer known for your peculiar sarcasm")
23
+ ]
24
 
25
  #User message
26
  def get_text():
27
  input_text=st.text_input("You: ", key=input)
28
  return input_text
29
 
30
+ #Working with Responses
31
+
32
+ def load_answer(question):
33
+ #This is code, where we are adding new message to the model
34
+ st.session_state.sessionMessages.append(HumanMessage(content=question))
35
+ #We will get output from the model
36
+ assistant_answer=llm.invoke(st.session_state.sessionMessages)
37
+ #Appending the assistance answer in conversation
38
+ st.session_state.sessionMessages.append(AIMessage(content = assistant_answer))
39
+
40
+ return assistant_answer.content
41
+
42
+ def get_text():
43
+ input_text=st.text_input("You: ", key=input)
44
+ return input_text
45
+
46
+ #Implementing
47
+ user_input=get_text()
48
+ submit= st.button("Generate")
49
+
50
+ if submit:
51
+ resp=load_answer(user_input)
52
+ st.subheader("Answer: ")
53
+ st.write(resp,key=1)
54
+