Spaces:

Pushpak21
/

DCM_Viewer

Sleeping

App Files Files Community

Pushpak21 commited on Jun 26, 2025

Commit

8161707

verified ·

1 Parent(s): 6f33aaf

Upload folder using huggingface_hub

Browse files

Files changed (4) hide show

Dockerfile +18 -0
app.py +90 -0
index.html +101 -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,101 @@

+<!-- index.html -->
+<!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>
+  <input type="file" id="file1" accept=".txt"> File A<br>
+  <input type="file" id="file2" accept=".txt"> File B<br><br>
+  <button onclick="uploadFiles()">Visualize</button>
+  <div id="plot"></div>
+  <script>
+    async function uploadFiles() {
+      const file1 = document.getElementById("file1").files[0];
+      const file2 = document.getElementById("file2").files[0];
+      if (!file1 || !file2) {
+        alert("Please select two files.");
+        return;
+      }
+      const data1 = await sendFile(file1);
+      const data2 = await sendFile(file2);
+      draw(data1, data2);
+    }
+    async function sendFile(file) {
+      const formData = new FormData();
+      formData.append("file", file);
+      const res = await fetch("http://localhost:7860/parse", {
+        method: "POST",
+        body: formData
+      });
+      return await res.json();
+    }
+    function draw(data1, data2) {
+      const traces = [];
+      function addAssembly(data, colorBase, colorNormal, prefix) {
+        for (const [tag, g] of Object.entries(data.geometry)) {
+          const [x, y, z] = g.base_point;
+          const [nx, ny, nz] = g.normal;
+          traces.push({
+            type: 'scatter3d',
+            mode: 'markers+text',
+            x: [x], y: [y], z: [z],
+            marker: { color: colorBase, size: 5 },
+            text: [`${prefix} ${g.label || tag}`],
+            textposition: 'top center'
+          });
+          traces.push({
+            type: 'scatter3d',
+            mode: 'lines',
+            x: [x, x + nx],
+            y: [y, y + ny],
+            z: [z, z + nz],
+            line: { color: colorNormal, width: 3 }
+          });
+        }
+        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;
+            traces.push({
+              type: 'scatter3d',
+              mode: 'lines',
+              x: [x1, x2], y: [y1, y2], z: [z1, z2],
+              line: { color: 'green', dash: 'dot', width: 2 }
+            });
+          }
+        }
+      }
+      addAssembly(data1, 'red', 'blue', 'File A');
+      addAssembly(data2, 'purple', 'orange', 'File B');
+      Plotly.newPlot('plot', traces, {
+        margin: { t: 30 },
+        scene: { aspectmode: 'data' },
+        title: '3D Assembly Viewer'
+      });
+    }
+  </script>
+</body>
+</html>

requirements.txt ADDED Viewed

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

+fastapi
+uvicorn
+python-multipart