Update app.py

app.py

@@ -1,52 +1,44 @@
-import os
-import sys
-import zipfile
-import shutil
-import subprocess
-
+import os, sys, zipfile, shutil, subprocess
 import gradio as gr
 
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-# This zip was uploaded to your space
 ZIP_PATH = os.path.join(BASE_DIR, "engineering_drawing_code.zip")
-# Where the repo will live once extracted
 REPO_DIR = os.path.join(BASE_DIR, "engineering_drawing_repo")
 
 def extract_two_outputs(uploaded_image_path):
-    # 1) Ensure the repo is unpacked
+    # 1) Unzip the extractor repo if needed
     if not os.path.isdir(REPO_DIR):
         with zipfile.ZipFile(ZIP_PATH, "r") as z:
             z.extractall(REPO_DIR)
-    # 2) Clean up any prior run
+    # 2) Clean previous outputs
     out_excel = os.path.join(BASE_DIR, "drawingInfo.xlsx")
     out_png   = os.path.join(BASE_DIR, "extracted", "drawing01.png")
     if os.path.exists(out_excel):
         os.remove(out_excel)
     if os.path.isdir(os.path.join(BASE_DIR, "extracted")):
         shutil.rmtree(os.path.join(BASE_DIR, "extracted"))
-    # 3) Prepare images/01.png for the OCR script
+    # 3) Prepare images folder
     images_folder = os.path.join(REPO_DIR, "images")
     if os.path.isdir(images_folder):
         shutil.rmtree(images_folder)
     os.makedirs(images_folder, exist_ok=True)
-    # Copy the user’s upload to images/01.png
-    shutil.copyfile(uploaded_image_path, os.path.join(images_folder, "01.png"))
-    # 4) Run the extraction script
+    shutil.copyfile(uploaded_image_path,
+                    os.path.join(images_folder, "01.png"))
+    # 4) Run the OCR/extraction script
     subprocess.run(
         [sys.executable, "mainExtractionOCR.py"],
         cwd=REPO_DIR,
         check=True,
     )
-    # 5) Return the two artifacts
+    # 5) Return the ballooned drawing + Excel
     return out_png, out_excel
 
-# Build Gradio interface
+# Build Gradio UI
 demo = gr.Blocks()
-
 with demo:
-    gr.Markdown("## Engineering-Drawing Extractor")
+    gr.Markdown("## Engineering-Drawing Information Extractor")
     with gr.Row():
-        inp = gr.Image(type="filepath", label="Upload your drawing")
+        inp = gr.Image(type="filepath", label="Upload drawing")
     with gr.Row():
         balloon = gr.Image(type="filepath", label="Ballooned Drawing")
         excel   = gr.File(label="Download Extracted Data (Excel)")