display inovonics logo

app.py

@@ -34,6 +34,14 @@ os.makedirs(UPLOAD_DIR, exist_ok=True)
 os.makedirs(JSON_DIR, exist_ok=True)
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
+# Copy logo.png to /tmp
+logo_src = os.path.join(BASE_DIR, "public", "logo.png")
+logo_dst = "/tmp/logo.png"
+
+# Only copy if not already there
+if not os.path.exists(logo_dst):
+    shutil.copy(logo_src, logo_dst)
+
 # Download model if missing
 if not os.path.exists(model_path):
     print("🚀 Model file not found! Downloading...")
@@ -95,12 +103,12 @@ with gr.Blocks() as demo:
     # Header
     with gr.Row():
         gr.Markdown(
-            """
-            <div style='display: flex; align-items: center; justify-content: center;'>
-                <img src='public/logo.png' style='height: 50px; margin-right: 10px;'/>
-                <h1>Inovonics 2D Floorplan Vectorizer</h1>
-            </div>
-            """
+            f"""
+    <div style='display: flex; align-items: center; justify-content: center;'>
+        <img src='file/tmp/logo.png' style='height: 50px; margin-right: 10px;'/>
+        <h1>Inovonics 2D Floorplan Vectorizer</h1>
+    </div>
+    """
         )
 
     with gr.Row():