Update app.py

app.py

@@ -2,47 +2,60 @@ import gradio as gr
 import tempfile
 import os
 import subprocess
+import uuid
 
 def process_files(pdf_file, word_file):
-    # 1. Save the uploaded files to temp
-    temp_dir = tempfile.mkdtemp()
+    # Each upload returns a path (str) with type="filepath"
+    # Create a unique temp directory for each run (prevents parallel collision)
+    temp_dir = tempfile.mkdtemp(prefix="hf_redtext_")
+
+    # Copy user-uploaded files into temp directory with standard names
     pdf_path = os.path.join(temp_dir, "input.pdf")
     word_path = os.path.join(temp_dir, "input.docx")
-    with open(pdf_path, "wb") as f:
-        f.write(pdf_file.read())
-    with open(word_path, "wb") as f:
-        f.write(word_file.read())
+    os.rename(pdf_file, pdf_path)
+    os.rename(word_file, word_path)
 
-    # 2. Step 1: Extract PDF data to txt
+    # Step 1: Extract PDF data to txt
     pdf_txt_path = os.path.join(temp_dir, "pdf_data.txt")
-    subprocess.run(["python", "extract_pdf_data.py", pdf_path, pdf_txt_path], check=True)
+    subprocess.run(
+        ["python", "extract_pdf_data.py", pdf_path, pdf_txt_path],
+        check=True
+    )
 
-    # 3. Step 2: Extract red text from Word to JSON
+    # Step 2: Extract red text from Word to JSON
     word_json_path = os.path.join(temp_dir, "word_data.json")
-    subprocess.run(["python", "extract_red_text.py", word_path, word_json_path], check=True)
+    subprocess.run(
+        ["python", "extract_red_text.py", word_path, word_json_path],
+        check=True
+    )
 
-    # 4. Step 3: Update docx JSON with PDF txt, output updated JSON
+    # Step 3: Update docx JSON with PDF txt, output updated JSON
     updated_json_path = os.path.join(temp_dir, "updated_word_data.json")
-    subprocess.run([
-        "python", "update_docx_with_pdf.py", word_json_path, pdf_txt_path, updated_json_path
-    ], check=True)
+    subprocess.run(
+        ["python", "update_docx_with_pdf.py", word_json_path, pdf_txt_path, updated_json_path],
+        check=True
+    )
 
-    # 5. Step 4: Compare word file with updated JSON and update docx
+    # Step 4: Compare word file with updated JSON and update docx
     final_docx_path = os.path.join(temp_dir, "updated.docx")
-    subprocess.run([
-        "python", "updated_word.py", word_path, updated_json_path, final_docx_path
-    ], check=True)
+    subprocess.run(
+        ["python", "updated_word.py", word_path, updated_json_path, final_docx_path],
+        check=True
+    )
 
-    # 6. Return final docx file
+    # Return final updated docx file
     return final_docx_path
 
-gr.Interface(
+iface = gr.Interface(
     fn=process_files,
     inputs=[
-        gr.File(label="Upload PDF File"),
-        gr.File(label="Upload Word File")
+        gr.File(label="Upload PDF File", type="filepath"),
+        gr.File(label="Upload Word File", type="filepath"),
     ],
     outputs=gr.File(label="Download Updated Word File"),
     title="Red Text Replacer",
     description="Upload a PDF and Word document. Red-colored text in the Word doc will be replaced by matching content from the PDF."
-).launch()
+)
+
+if __name__ == "__main__":
+    iface.launch()