Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -138,20 +138,12 @@ def main():
|
|
138 |
# Get response
|
139 |
response = qa_chain.invoke({'query': prompt})
|
140 |
|
141 |
-
# Extract result
|
142 |
result = response["result"]
|
143 |
-
source_documents = response["source_documents"]
|
144 |
|
145 |
-
#
|
146 |
-
|
147 |
-
|
148 |
-
source_docs_text += f"{i}. Page {doc.metadata.get('page', 'N/A')}: {doc.page_content[:100]}...\n\n"
|
149 |
-
|
150 |
-
# Display result and sources
|
151 |
-
result_to_show = f"{result}\n{source_docs_text}"
|
152 |
-
|
153 |
-
st.chat_message('assistant').markdown(result_to_show)
|
154 |
-
st.session_state.messages.append({'role': 'assistant', 'content': result_to_show})
|
155 |
|
156 |
except Exception as e:
|
157 |
error_message = f"Error: {str(e)}"
|
@@ -160,4 +152,4 @@ def main():
|
|
160 |
st.session_state.messages.append({'role': 'assistant', 'content': error_message})
|
161 |
|
162 |
if __name__ == "__main__":
|
163 |
-
main()
|
|
|
138 |
# Get response
|
139 |
response = qa_chain.invoke({'query': prompt})
|
140 |
|
141 |
+
# Extract result only (no sources)
|
142 |
result = response["result"]
|
|
|
143 |
|
144 |
+
# Display only the result
|
145 |
+
st.chat_message('assistant').markdown(result)
|
146 |
+
st.session_state.messages.append({'role': 'assistant', 'content': result})
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
147 |
|
148 |
except Exception as e:
|
149 |
error_message = f"Error: {str(e)}"
|
|
|
152 |
st.session_state.messages.append({'role': 'assistant', 'content': error_message})
|
153 |
|
154 |
if __name__ == "__main__":
|
155 |
+
main()
|