vishwask commited on
Commit
6b25d53
·
verified ·
1 Parent(s): c64d65e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -1
app.py CHANGED
@@ -158,4 +158,30 @@ def conversation(qa_chain, message, history):
158
  new_history = history + [(message, response_answer)]
159
  # return gr.update(value=""), new_history, response_sources[0], response_sources[1]
160
  return qa_chain, gr.update(value=""), new_history, response_source1, response_source1_page, response_source2, response_source2_page, response_source3, response_source3_page
161
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
158
  new_history = history + [(message, response_answer)]
159
  # return gr.update(value=""), new_history, response_sources[0], response_sources[1]
160
  return qa_chain, gr.update(value=""), new_history, response_source1, response_source1_page, response_source2, response_source2_page, response_source3, response_source3_page
161
+
162
+
163
+
164
+ def demo():
165
+ with gr.Blocks(theme="base") as demo:
166
+ vector_db = gr.State()
167
+ qa_chain = gr.State()
168
+ collection_name = gr.State()
169
+
170
+ with gr.Tab("Chat"):
171
+ chatbot = gr.Chatbot(height=300)
172
+ with gr.Accordion("Advanced - Document references", open=False):
173
+ with gr.Row():
174
+ doc_source1 = gr.Textbox(label="Reference 1", lines=2, container=True, scale=20)
175
+ source1_page = gr.Number(label="Page", scale=1)
176
+ with gr.Row():
177
+ doc_source2 = gr.Textbox(label="Reference 2", lines=2, container=True, scale=20)
178
+ source2_page = gr.Number(label="Page", scale=1)
179
+ with gr.Row():
180
+ doc_source3 = gr.Textbox(label="Reference 3", lines=2, container=True, scale=20)
181
+ source3_page = gr.Number(label="Page", scale=1)
182
+ with gr.Row():
183
+ msg = gr.Textbox(placeholder="Type message", container=True)
184
+ with gr.Row():
185
+ submit_btn = gr.Button("Submit")
186
+ clear_btn = gr.ClearButton([msg, chatbot])
187
+