Spaces:
Sleeping
Sleeping
Beta-version
Browse files
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 |
+
|