Spaces:
Sleeping
Sleeping
Abid Ali Awan
commited on
Commit
·
2987ac4
1
Parent(s):
5ca6388
added qa part
Browse files
app.py
CHANGED
|
@@ -62,8 +62,8 @@ def load_files(file_path: str):
|
|
| 62 |
# Respond function
|
| 63 |
def respond(message, history):
|
| 64 |
# Use the preloaded LLM
|
| 65 |
-
query_engine = vector_index.
|
| 66 |
-
streaming_response = query_engine.
|
| 67 |
partial_text = ""
|
| 68 |
for new_text in streaming_response.response_gen:
|
| 69 |
partial_text += new_text
|
|
@@ -71,6 +71,13 @@ def respond(message, history):
|
|
| 71 |
yield partial_text
|
| 72 |
|
| 73 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 74 |
# UI Setup
|
| 75 |
with gr.Blocks(
|
| 76 |
theme=gr.themes.Default(
|
|
@@ -105,7 +112,7 @@ with gr.Blocks(
|
|
| 105 |
# Set up Gradio interactions
|
| 106 |
btn.click(fn=load_files, inputs=file_input, outputs=output)
|
| 107 |
clear.click(
|
| 108 |
-
|
| 109 |
outputs=[file_input, output],
|
| 110 |
)
|
| 111 |
|
|
|
|
| 62 |
# Respond function
|
| 63 |
def respond(message, history):
|
| 64 |
# Use the preloaded LLM
|
| 65 |
+
query_engine = vector_index.as_query_engine(streaming=True, llm=llm)
|
| 66 |
+
streaming_response = query_engine.query(message)
|
| 67 |
partial_text = ""
|
| 68 |
for new_text in streaming_response.response_gen:
|
| 69 |
partial_text += new_text
|
|
|
|
| 71 |
yield partial_text
|
| 72 |
|
| 73 |
|
| 74 |
+
# Clear function
|
| 75 |
+
def clear_state():
|
| 76 |
+
global vector_index
|
| 77 |
+
vector_index = None
|
| 78 |
+
return [None, None, None]
|
| 79 |
+
|
| 80 |
+
|
| 81 |
# UI Setup
|
| 82 |
with gr.Blocks(
|
| 83 |
theme=gr.themes.Default(
|
|
|
|
| 112 |
# Set up Gradio interactions
|
| 113 |
btn.click(fn=load_files, inputs=file_input, outputs=output)
|
| 114 |
clear.click(
|
| 115 |
+
fn=clear_state, # Use the clear_state function
|
| 116 |
outputs=[file_input, output],
|
| 117 |
)
|
| 118 |
|