Spaces:

Pushpak21
/

DCM_Viewer

Sleeping

App Files Files Community

Pushpak21 commited on Jun 26, 2025

Commit

5c8d4fd

verified ·

1 Parent(s): e93edf5

Upload folder using huggingface_hub

Browse files

Files changed (4) hide show

Dockerfile +18 -0
app.py +90 -0
index.html +85 -0
requirements.txt +4 -0

Dockerfile ADDED Viewed

	@@ -0,0 +1,18 @@

+# Dockerfile
+FROM python:3.11
+# Install backend dependencies
+COPY requirements.txt /app/requirements.txt
+WORKDIR /app
+RUN pip install -r requirements.txt
+# Copy backend and frontend
+COPY app.py /app/app.py
+COPY index.html /app/index.html
+# Add CORS and serve index.html statically
+RUN mkdir -p /app/static
+# Run both backend (FastAPI) and static file server
+CMD uvicorn app:app --host 0.0.0.0 --port 7860

app.py ADDED Viewed

	@@ -0,0 +1,90 @@

+from fastapi import FastAPI, UploadFile, File
+from fastapi.responses import JSONResponse, FileResponse
+from fastapi.middleware.cors import CORSMiddleware
+import os
+app = FastAPI()
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"], allow_credentials=True,
+    allow_methods=["*"], allow_headers=["*"],
+)
+@app.get("/")
+def serve_frontend():
+    return FileResponse("index.html")
+@app.post("/parse")
+async def parse_file(file: UploadFile = File(...)):
+    content = await file.read()
+    return JSONResponse(content=parse_dcm_assembly(content))
+def parse_dcm_assembly(file_bytes):
+    lines = file_bytes.decode("utf-8").splitlines()
+    blocks, current = [], []
+    for line in lines:
+        if line.startswith("CLASS"):
+            if current:
+                blocks.append(current)
+                current = []
+        current.append(line.strip())
+    if current: blocks.append(current)
+    parts, geometry, constraints = [], {}, []
+    plane_map, tag_to_plane, label_map = {}, {}, {}
+    for block in blocks:
+        head = block[0]
+        if "CLASS SET " in head:
+            part, members = {}, []
+            for line in block:
+                if "DCM3_TEXT_LABEL" in line:
+                    part['name'] = line.split("DCM3_TEXT_LABEL", 1)[1].strip()
+                elif "FMEMBER TAGS" in line:
+                    members = list(map(int, line.split()[2:]))
+            if 'name' in part:
+                part['member_tags'] = members
+                parts.append(part)
+        elif "CLASS PLANE " in head:
+            pid = int(head.split()[-1])
+            base, norm = None, None
+            for line in block:
+                if "FBASE_POINT" in line:
+                    base = tuple(map(float, line.split()[1:]))
+                elif "FNORMAL" in line:
+                    norm = tuple(map(float, line.split()[1:]))
+            if base and norm:
+                plane_map[pid] = {'base_point': base, 'normal': norm}
+        elif "CLASS WORK_PLANE " in head:
+            wid = int(head.split()[-1])
+            tag, label = None, None
+            for line in block:
+                if "FMY_PLANE TAG" in line:
+                    tag = int(line.split()[-1])
+                elif "DCM3_TEXT_LABEL" in line:
+                    label = line.split("DCM3_TEXT_LABEL", 1)[1].strip()
+            if tag:
+                tag_to_plane[wid] = tag
+                label_map[wid] = label
+        elif "CLASS CONSTRAINT " in head:
+            con = {}
+            for line in block:
+                if "FEND_1 TAG" in line:
+                    con['from'] = int(line.split()[-1])
+                elif "FEND_2 TAG" in line:
+                    con['to'] = int(line.split()[-1])
+                elif "DCM3_TEXT_LABEL" in line:
+                    con['label'] = line.split("DCM3_TEXT_LABEL", 1)[1].strip()
+            if con: constraints.append(con)
+    for wid, pid in tag_to_plane.items():
+        if pid in plane_map:
+            geometry[wid] = {**plane_map[pid], 'label': label_map.get(wid)}
+    return { "parts": parts, "geometry": geometry, "constraints": constraints }

index.html ADDED Viewed

	@@ -0,0 +1,85 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8" />
+  <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+  <title>DCM Assembly Visualizer</title>
+  <script src="https://cdn.plot.ly/plotly-latest.min.js"></script>
+  <style>
+    body { font-family: sans-serif; padding: 20px; }
+    #plot { width: 100%; height: 700px; }
+  </style>
+</head>
+<body>
+  <h1>DCM Assembly Visualizer</h1>
+  <p>This visualizer is now intended to be used with a Streamlit frontend for file upload.</p>
+  <div id="plot"></div>
+  <script>
+    let allTraces = [];
+    let normalTraces = [];
+    async function drawFromServer() {
+      const res = await fetch("/parsed_data.json");
+      const payload = await res.json();
+      allTraces = [];
+      normalTraces = [];
+      for (const [color, label, data] of payload.files) {
+        extractTraces(data, color, label);
+      }
+      Plotly.newPlot('plot', [...allTraces, ...normalTraces.filter(t => t.visible)], {
+        margin: { t: 30 },
+        scene: { aspectmode: 'data' },
+        title: '3D Assembly Viewer'
+      });
+    }
+    function extractTraces(data, baseColor, normalColor) {
+      for (const [tag, g] of Object.entries(data.geometry)) {
+        const [x, y, z] = g.base_point;
+        const [nx, ny, nz] = g.normal;
+        allTraces.push({
+          type: 'scatter3d',
+          mode: 'markers+text',
+          x: [x], y: [y], z: [z],
+          marker: { color: baseColor, size: 5 },
+          text: [g.label || tag],
+          textposition: 'top center'
+        });
+        normalTraces.push({
+          type: 'scatter3d',
+          mode: 'lines',
+          x: [x, x + nx],
+          y: [y, y + ny],
+          z: [z, z + nz],
+          line: { color: normalColor, width: 2 },
+          visible: true
+        });
+      }
+      for (const c of data.constraints) {
+        const g1 = data.geometry[c.from];
+        const g2 = data.geometry[c.to];
+        if (g1 && g2) {
+          const [x1, y1, z1] = g1.base_point;
+          const [x2, y2, z2] = g2.base_point;
+          allTraces.push({
+            type: 'scatter3d',
+            mode: 'lines',
+            x: [x1, x2], y: [y1, y2], z: [z1, z2],
+            line: { color: 'green', dash: 'dot', width: 2 }
+          });
+        }
+      }
+    }
+    drawFromServer();
+  </script>
+</body>
+</html>

requirements.txt ADDED Viewed

	@@ -0,0 +1,4 @@

+fastapi
+uvicorn
+python-multipart