Update app.py
Browse files
app.py
CHANGED
|
@@ -135,26 +135,30 @@ with gr.Blocks() as demo:
|
|
| 135 |
api_btn = gr.Button("Set API Key", elem_classes=["primary"])
|
| 136 |
api_status = gr.Markdown()
|
| 137 |
|
| 138 |
-
|
| 139 |
-
|
| 140 |
-
|
| 141 |
-
|
| 142 |
-
|
| 143 |
-
|
| 144 |
|
| 145 |
gr.Markdown("---")
|
|
|
|
| 146 |
gr.Markdown("### 2) Ask questions about your file")
|
| 147 |
question_tb = gr.Textbox(label="Your question", placeholder="Type a question…")
|
| 148 |
ask_btn = gr.Button("Ask", elem_classes=["primary"])
|
| 149 |
chatbot = gr.Chatbot()
|
| 150 |
|
|
|
|
| 151 |
api_btn.click(ui_set_api_key, [api_tb], [client_state, api_status])
|
|
|
|
| 152 |
upload_btn.click(
|
| 153 |
upload_and_index,
|
| 154 |
[client_state, file_uploader],
|
| 155 |
[store_state, upload_status, upload_btn],
|
| 156 |
show_progress=True
|
| 157 |
)
|
|
|
|
| 158 |
ask_btn.click(
|
| 159 |
ask,
|
| 160 |
[client_state, store_state, chat_state, question_tb, gr.State(DEFAULT_MODEL)],
|
|
|
|
| 135 |
api_btn = gr.Button("Set API Key", elem_classes=["primary"])
|
| 136 |
api_status = gr.Markdown()
|
| 137 |
|
| 138 |
+
# Upload section
|
| 139 |
+
gr.Markdown("### 1) Upload & Index your file")
|
| 140 |
+
with gr.Row():
|
| 141 |
+
file_uploader = gr.File(label="Choose file to upload", file_types=['.txt', '.pdf', '.docx'])
|
| 142 |
+
upload_btn = gr.Button("Upload & Index", elem_classes=["primary"])
|
| 143 |
+
upload_status = gr.Markdown()
|
| 144 |
|
| 145 |
gr.Markdown("---")
|
| 146 |
+
# Chat section
|
| 147 |
gr.Markdown("### 2) Ask questions about your file")
|
| 148 |
question_tb = gr.Textbox(label="Your question", placeholder="Type a question…")
|
| 149 |
ask_btn = gr.Button("Ask", elem_classes=["primary"])
|
| 150 |
chatbot = gr.Chatbot()
|
| 151 |
|
| 152 |
+
# Define interactions
|
| 153 |
api_btn.click(ui_set_api_key, [api_tb], [client_state, api_status])
|
| 154 |
+
|
| 155 |
upload_btn.click(
|
| 156 |
upload_and_index,
|
| 157 |
[client_state, file_uploader],
|
| 158 |
[store_state, upload_status, upload_btn],
|
| 159 |
show_progress=True
|
| 160 |
)
|
| 161 |
+
|
| 162 |
ask_btn.click(
|
| 163 |
ask,
|
| 164 |
[client_state, store_state, chat_state, question_tb, gr.State(DEFAULT_MODEL)],
|