Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -98,11 +98,18 @@ class PDFQAProcessor:
|
|
| 98 |
|
| 99 |
response = rag_chain.invoke({"input": question})
|
| 100 |
|
| 101 |
-
final_response = response["answer"] + "\n\nSources\n\n"
|
| 102 |
|
| 103 |
-
for info in response["context"]:
|
| 104 |
-
|
| 105 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 106 |
return final_response
|
| 107 |
except Exception as e:
|
| 108 |
return f"Error answering question: {str(e)}", None
|
|
@@ -117,7 +124,8 @@ with gr.Blocks(title="PDF QA Assistant") as demo:
|
|
| 117 |
|
| 118 |
with gr.Tab("Ask Questions"):
|
| 119 |
question_input = gr.Textbox(label="Your Question")
|
| 120 |
-
answer_output = gr.Textbox(label="Answer", interactive=False)
|
|
|
|
| 121 |
ask_btn = gr.Button("Ask Question")
|
| 122 |
|
| 123 |
process_btn.click(
|
|
|
|
| 98 |
|
| 99 |
response = rag_chain.invoke({"input": question})
|
| 100 |
|
| 101 |
+
# final_response = response["answer"] + "\n\nSources\n\n"
|
| 102 |
|
| 103 |
+
# for info in response["context"]:
|
| 104 |
+
# final_response += info.page_content + "\nSource of Info: " + info.metadata['source'] + "\nAt Page No: " + info.metadata['page_label']+"\n\n"
|
| 105 |
|
| 106 |
+
final_response = response["answer"] + "\n\n### Sources\n\n" # Changed to use markdown formatting
|
| 107 |
+
for info in response["context"]:
|
| 108 |
+
final_response += (
|
| 109 |
+
f"{info.page_content}<br>" # Changed to use markdown bold formatting
|
| 110 |
+
f"Source of Info: {info.metadata['source']}<br>"
|
| 111 |
+
f"At Page No: {info.metadata['page_label']}<br><br>"
|
| 112 |
+
)
|
| 113 |
return final_response
|
| 114 |
except Exception as e:
|
| 115 |
return f"Error answering question: {str(e)}", None
|
|
|
|
| 124 |
|
| 125 |
with gr.Tab("Ask Questions"):
|
| 126 |
question_input = gr.Textbox(label="Your Question")
|
| 127 |
+
# answer_output = gr.Textbox(label="Answer", interactive=False)
|
| 128 |
+
answer_output = gr.Markdown(label="Answer")
|
| 129 |
ask_btn = gr.Button("Ask Question")
|
| 130 |
|
| 131 |
process_btn.click(
|