Update app.py

app.py

@@ -1,12 +1,13 @@
-import streamlit as st
-import os
+from fastapi import FastAPI, UploadFile, File
+from fastapi.responses import FileResponse, JSONResponse
+from fastapi.staticfiles import StaticFiles
 import json
 
-st.set_page_config(page_title="DCM Assembly Visualizer", layout="wide")
-st.title("DCM Assembly Visualizer")
+app = FastAPI()
+app.mount("/", StaticFiles(directory="static", html=True), name="static")
 
 def parse_dcm_assembly(uploaded_file):
-    lines = uploaded_file.read().decode("utf-8").splitlines()
+    lines = uploaded_file.file.read().decode("utf-8").splitlines()
 
     geometry = {}
     parts = []
@@ -16,19 +17,16 @@ def parse_dcm_assembly(uploaded_file):
     temp_part = {}
     temp_constraint = {}
     current_class = None
-    current_id = None
 
-    plane_label_map = {}   # tag → label (from WORK_PLANE)
-    plane_point_map = {}   # tag → base point + normal
+    plane_label_map = {}
+    plane_point_map = {}
 
     for line in lines:
         line = line.strip()
-
         if line.startswith("CLASS"):
             parts_line = line.split()
             current_class = parts_line[1]
             current_id = int(parts_line[2])
-
             if current_class == "PLANE":
                 temp_geom = {"tag": current_id}
             elif current_class == "WORK_PLANE":
@@ -72,7 +70,6 @@ def parse_dcm_assembly(uploaded_file):
                 if "from" in temp_constraint and "to" in temp_constraint:
                     constraints[(temp_constraint["from"], temp_constraint["to"])] = temp_constraint["label"]
 
-    # Build final geometry dict
     final_geometry = {}
     for tag, data in plane_point_map.items():
         final_geometry[tag] = {
@@ -81,7 +78,6 @@ def parse_dcm_assembly(uploaded_file):
             "label": plane_label_map.get(tag, "")
         }
 
-    # Build final constraint list
     constraint_list = [{"from": f, "to": t, "label": lbl} for (f, t), lbl in constraints.items()]
 
     return {
@@ -90,26 +86,9 @@ def parse_dcm_assembly(uploaded_file):
         "constraints": constraint_list
     }
 
-
-# === File Upload ===
-file1 = st.file_uploader("Upload File A", type=["txt"])
-file2 = st.file_uploader("Upload File B", type=["txt"])
-
-parsed_files = []
-
-if file1:
-    parsed_files.append(("red", "blue", parse_dcm_assembly(file1)))
-if file2:
-    parsed_files.append(("purple", "orange", parse_dcm_assembly(file2)))
-
-# === Draw Button + Iframe Viewer ===
-if parsed_files:
-    if st.button("🎯 Visualize"):
-        os.makedirs("static", exist_ok=True)
-        with open("static/parsed_data.json", "w") as f:
-            json.dump({"files": parsed_files}, f, indent=2)
-
-        st.components.v1.html(
-            '<iframe src="static/index.html" width="100%" height="800" style="border:none;"></iframe>',
-            height=800,
-        )
+@app.post("/upload")
+async def upload_file(file: UploadFile = File(...)):
+    data = parse_dcm_assembly(file)
+    with open("static/parsed_data.json", "w") as f:
+        json.dump({"files": [("red", "blue", data)]}, f)
+    return JSONResponse(content={"status": "ok"})