Somnath3570 commited on
Commit
69c5665
·
verified ·
1 Parent(s): 97f4a8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -13
app.py CHANGED
@@ -138,20 +138,12 @@ def main():
138
  # Get response
139
  response = qa_chain.invoke({'query': prompt})
140
 
141
- # Extract result and sources
142
  result = response["result"]
143
- source_documents = response["source_documents"]
144
 
145
- # Format source documents
146
- source_docs_text = "\n\n**Sources:**\n"
147
- for i, doc in enumerate(source_documents, 1):
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()