astro21 commited on
Commit
8566bc6
·
verified ·
1 Parent(s): 1b6cc04

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from fastapi.middleware.cors import CORSMiddleware
4
  from typing import List
5
  from langchain_community.chat_models import ChatOpenAI
6
  from utils import process_file_with_dedoc, extract_text_from_all_levels, generate_formatted_resume, \
7
- generate_json_structured_resume
8
  import shutil
9
  import os
10
 
@@ -54,6 +54,11 @@ async def parse_resume(files: List[UploadFile] = File(...)):
54
  "parsed_resume": parsed_resume,
55
  "parsed_json_resume": parsed_json_resume
56
  })
 
 
 
 
 
57
  else:
58
  return JSONResponse(status_code=400, content={
59
  "message": "Invalid file type. Allowed file types are: jpg, jpeg, png, docx, pdf, html, doc"})
 
4
  from typing import List
5
  from langchain_community.chat_models import ChatOpenAI
6
  from utils import process_file_with_dedoc, extract_text_from_all_levels, generate_formatted_resume, \
7
+ generate_json_structured_resume,
8
  import shutil
9
  import os
10
 
 
54
  "parsed_resume": parsed_resume,
55
  "parsed_json_resume": parsed_json_resume
56
  })
57
+
58
+ # Delete the uploaded file after processing
59
+ os.remove("/temp_files/"+uploaded_file.filename)
60
+ print(f"Deleted file: {uploaded_file.filename}")
61
+
62
  else:
63
  return JSONResponse(status_code=400, content={
64
  "message": "Invalid file type. Allowed file types are: jpg, jpeg, png, docx, pdf, html, doc"})