iShare commited on
Commit
e269e3e
·
1 Parent(s): 2af65fc

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +9 -9
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(username: 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("username")
83
  filename = request.query_params.get("filename")
84
- print(username)
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
- #return {"INFO": f"File '{uploaded_file.filename}' saved to your profile."}
154
- #return api_call_msg
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})