Update app.py
Browse files
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 |
+
|