add backend files

backend/app.py

@@ -22,4 +22,8 @@ def greet_json():
 
 # Serve frontend build under / (but exclude /api)
 frontend_path = Path(__file__).parent.parent / "frontend" / "dist"
-app.mount("/", StaticFiles(directory=frontend_path, html=True), name="frontend")
+
+if frontend_path.exists():
+    app.mount("/", StaticFiles(directory=frontend_path, html=True), name="frontend")
+else:
+    print("Frontend build not found. Skipping StaticFiles mount.")

backend/hf_model_utils.py

@@ -0,0 +1,63 @@
+import torch
+import torch.nn as nn
+import json
+import hashlib
+import torch
+import gc
+from transformers import AutoConfig, AutoModel
+
+
+def module_hash(module):
+    """
+    Generate a hash representing the structure of a module.
+    Uses class name + child hashes + param shapes to detect repeats.
+    """
+    children = list(module.named_children())
+    child_hashes = []
+    for name, child in children:
+        child_hashes.append(module_hash(child))
+
+    # Include class name and param shapes
+    param_info = [(name, tuple(p.shape), p.requires_grad)
+                  for name, p in module.named_parameters(recurse=False)]
+
+    rep = (module.__class__.__name__, tuple(child_hashes), tuple(param_info))
+    rep_bytes = str(rep).encode('utf-8')
+    return hashlib.md5(rep_bytes).hexdigest()
+
+
+def hf_style_structural_dict(module):
+    """
+    Recursively convert a PyTorch module into a dict mirroring
+    Hugging Face's print(model), only counting repeats when structure is identical.
+    """
+    children = list(module.named_children())
+    result = {"class_name": module.__class__.__name__}
+
+    # Include params if present
+    params = {name: {"shape": list(p.shape), "requires_grad": p.requires_grad}
+              for name, p in module.named_parameters(recurse=False)}
+    if params:
+        result["params"] = params
+
+    if children:
+        child_dict = {}
+        i = 0
+        while i < len(children):
+            name, child = children[i]
+            current_hash = module_hash(child)
+            count = 1
+            j = i + 1
+            # Count consecutive children that are structurally identical
+            while j < len(children) and module_hash(children[j][1]) == current_hash:
+                count += 1
+                j += 1
+
+            child_entry = hf_style_structural_dict(child)
+            if count > 1:
+                child_entry["num_repeats"] = count
+            child_dict[name] = child_entry
+            i += count
+        result["children"] = child_dict
+
+    return result

backend/requirements.txt

@@ -1,2 +1,40 @@
-fastapi
-uvicorn[standard]
+annotated-types==0.7.0
+anyio==4.10.0
+certifi==2025.8.3
+charset-normalizer==3.4.3
+click==8.2.1
+fastapi==0.116.1
+filelock==3.19.1
+fsspec==2025.9.0
+h11==0.16.0
+hf-xet==1.1.10
+httptools==0.6.4
+idna==3.10
+Jinja2==3.1.6
+MarkupSafe==3.0.2
+mpmath==1.3.0
+networkx==3.5
+numpy==2.3.3
+packaging==25.0
+pydantic==2.11.9
+pydantic_core==2.33.2
+python-dotenv==1.1.1
+PyYAML==6.0.2
+regex==2025.9.1
+requests==2.32.5
+safetensors==0.6.2
+setuptools==80.9.0
+sniffio==1.3.1
+starlette==0.47.3
+sympy==1.14.0
+tokenizers==0.22.0
+torch==2.8.0
+tqdm==4.67.1
+transformers==4.56.1
+typing-inspection==0.4.1
+typing_extensions==4.15.0
+urllib3==2.5.0
+uvicorn==0.35.0
+uvloop==0.21.0
+watchfiles==1.1.0
+websockets==15.0.1

frontend/vite.config.js

@@ -5,7 +5,10 @@ export default defineConfig({
   plugins: [react()],
   server: {
     proxy: {
-      '/api': 'http://localhost:8000'  // forward /api requests to FastAPI
+      '/api': {
+        target: 'http://localhost:8000', // your FastAPI backend
+        changeOrigin: true,
+      },
     }
   }
 })