Update main.py

main.py

@@ -64,6 +64,15 @@ async def save_file(file: UploadFile) -> str:
         await out_file.write(content)
     return file_path
 
+# ✅ Extract text from DOCX
+def extract_text_from_docx(docx_path: str) -> str:
+    try:
+        doc = Document(docx_path)
+        text = "\n".join([para.text for para in doc.paragraphs])
+        return text.strip()
+    except Exception as e:
+        return f"Error extracting text from DOCX: {str(e)}"
+
 # ✅ Extract text from PDF using PyPDF2
 def extract_text_from_pdf(pdf_path: str) -> str:
     text = ""
@@ -86,11 +95,16 @@ async def parse_resume(file: UploadFile = File(...)):
         print(f"✅ File saved at {path}")
 
         print("📄 Extracting text...")
-        text = extract_text_from_pdf(path)
-        print("✅ Text extracted.")
+        ext = os.path.splitext(path)[-1].lower()
+        if ext == ".pdf":
+            text = extract_text_from_pdf(path)
+        elif ext in [".docx", ".doc"]:
+            text = extract_text_from_docx(path)
+        else:
+            os.remove(path)
+            return JSONResponse(status_code=400, content={"error": "Unsupported file type"})
 
-        json_result = parse_resume_text(text)
-        print("✅ JSON Created.")
+        print("✅ Text extracted.")
         
         os.remove(path)
         print("🧹 File removed.")