fahmiaziz98 commited on
Commit
7bc82ac
·
1 Parent(s): 7cd0cdf
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -24,7 +24,8 @@ if "retriever" not in st.session_state:
24
  st.session_state.retriever = None
25
  if "vector_store" not in st.session_state:
26
  st.session_state.vector_store = None
27
- workflow = None
 
28
 
29
 
30
  st.set_page_config(
@@ -95,8 +96,12 @@ if prompt := st.chat_input("Ask a question about your document"):
95
  ]
96
  }
97
 
98
- response = workflow.invoke(inputs)
99
- final_response = response["messages"][-1].content
 
 
 
 
100
 
101
  st.markdown(final_response)
102
  st.session_state.messages.append({"role": "assistant", "content": final_response})
 
24
  st.session_state.retriever = None
25
  if "vector_store" not in st.session_state:
26
  st.session_state.vector_store = None
27
+ if "workflow" not in st.session_state:
28
+ st.session_state.workflow = None
29
 
30
 
31
  st.set_page_config(
 
96
  ]
97
  }
98
 
99
+ # Generate response using workflow
100
+ if st.session_state.workflow is not None:
101
+ response = st.session_state.workflow.invoke(inputs)
102
+ final_response = response["messages"][-1].content
103
+ else:
104
+ final_response = "Please upload a PDF document first."
105
 
106
  st.markdown(final_response)
107
  st.session_state.messages.append({"role": "assistant", "content": final_response})