manabb commited on
Commit
c8e4e26
·
verified ·
1 Parent(s): 118ba9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -8
app.py CHANGED
@@ -89,14 +89,7 @@ def create_qa_chain(retriever):
89
 
90
 
91
  #=======================
92
- def chat(message, history):
93
- answer = qa_chain.invoke(message)
94
- # Get docs for refs
95
- docs = retriever.invoke(message)
96
- refs = [f"Page {d.metadata.get('page', 'N/A')}" for d in docs]
97
- full = f"{answer}\n\nRefs: {' | '.join(refs)}"
98
- history.append([message, full])
99
- return history, ""
100
  #============starting extract_docx_text
101
  def respond(message, history, doc_choice):
102
 
@@ -111,6 +104,7 @@ def respond(message, history, doc_choice):
111
  ]
112
  return "", new_history
113
  else:
 
114
  if doc_choice == "gpc_goods":
115
  retriever=retrieve_chunks_GPC()
116
  else:
 
89
 
90
 
91
  #=======================
92
+
 
 
 
 
 
 
 
93
  #============starting extract_docx_text
94
  def respond(message, history, doc_choice):
95
 
 
104
  ]
105
  return "", new_history
106
  else:
107
+ retriever=None
108
  if doc_choice == "gpc_goods":
109
  retriever=retrieve_chunks_GPC()
110
  else: