Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -74,22 +74,21 @@ async def home():
|
|
74 |
return "API Working!"
|
75 |
|
76 |
@app.post("/fastapi_file_upload_process")
|
77 |
-
#async def upload_file(
|
78 |
async def pdf_file_qa_process(username: str, request: Request, file_to_process: FileToProcess = Depends()):
|
79 |
uploaded_file = file_to_process.uploaded_file
|
80 |
print("File received:"+uploaded_file.filename)
|
81 |
|
82 |
-
username = request.query_params.get("
|
83 |
filename = request.query_params.get("filename")
|
84 |
-
print(
|
85 |
-
print(filename)
|
86 |
|
87 |
random_string = generate_random_string(20)
|
88 |
file_path = Path.cwd() / random_string
|
89 |
file_path.mkdir(parents=True, exist_ok=True)
|
90 |
file_saved_in_api = file_path / uploaded_file.filename
|
91 |
print(file_saved_in_api)
|
92 |
-
print(file_saved_in_api.resolve())
|
93 |
|
94 |
with open(file_saved_in_api, "wb+") as file_object:
|
95 |
file_object.write(uploaded_file.file.read())
|
@@ -125,6 +124,8 @@ async def pdf_file_qa_process(username: str, request: Request, file_to_process:
|
|
125 |
for i in range(len(hits[0])):
|
126 |
page_content = texts[hits[0][i]['corpus_id']]
|
127 |
page_contents.append(page_content)
|
|
|
|
|
128 |
|
129 |
temp_page_contents=str(page_contents)
|
130 |
final_page_contents = temp_page_contents.replace('\\n', '')
|
@@ -150,7 +151,6 @@ async def pdf_file_qa_process(username: str, request: Request, file_to_process:
|
|
150 |
new_final_ai_response = new_final_ai_response.split('Note:')[0].strip()
|
151 |
new_final_ai_response = new_final_ai_response.split('Please provide feedback on how to improve the chatbot.')[0].strip()
|
152 |
print(new_final_ai_response)
|
153 |
-
|
154 |
-
|
155 |
-
#return {"AIResponse": new_final_ai_response}
|
156 |
-
return JSONResponse({"AIResponse": new_final_ai_response})
|
|
|
74 |
return "API Working!"
|
75 |
|
76 |
@app.post("/fastapi_file_upload_process")
|
77 |
+
#async def upload_file(user_question: str, file_to_process: FileToProcess = Depends()):
|
78 |
async def pdf_file_qa_process(username: str, request: Request, file_to_process: FileToProcess = Depends()):
|
79 |
uploaded_file = file_to_process.uploaded_file
|
80 |
print("File received:"+uploaded_file.filename)
|
81 |
|
82 |
+
username = request.query_params.get("user_question")
|
83 |
filename = request.query_params.get("filename")
|
84 |
+
print("User entered question: "+user_question)
|
85 |
+
print("User uploaded file: "+filename)
|
86 |
|
87 |
random_string = generate_random_string(20)
|
88 |
file_path = Path.cwd() / random_string
|
89 |
file_path.mkdir(parents=True, exist_ok=True)
|
90 |
file_saved_in_api = file_path / uploaded_file.filename
|
91 |
print(file_saved_in_api)
|
|
|
92 |
|
93 |
with open(file_saved_in_api, "wb+") as file_object:
|
94 |
file_object.write(uploaded_file.file.read())
|
|
|
124 |
for i in range(len(hits[0])):
|
125 |
page_content = texts[hits[0][i]['corpus_id']]
|
126 |
page_contents.append(page_content)
|
127 |
+
|
128 |
+
print(page_contents)
|
129 |
|
130 |
temp_page_contents=str(page_contents)
|
131 |
final_page_contents = temp_page_contents.replace('\\n', '')
|
|
|
151 |
new_final_ai_response = new_final_ai_response.split('Note:')[0].strip()
|
152 |
new_final_ai_response = new_final_ai_response.split('Please provide feedback on how to improve the chatbot.')[0].strip()
|
153 |
print(new_final_ai_response)
|
154 |
+
|
155 |
+
return {"AIResponse": new_final_ai_response}
|
156 |
+
#return JSONResponse({"AIResponse": new_final_ai_response})
|
|