Fix the loading of static frontend assets

backend/app.py

@@ -19,10 +19,11 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Mount static files (frontend) so that visiting "/" serves index.html
-# The directory path "../frontend" works because when running in Docker,
-# our working directory is set to /app, and the frontend folder is at /app/frontend.
-app.mount("/static", StaticFiles(directory="frontend", html=True), name="static")
+# Mount static files for React app assets (CSS, JS)
+app.mount("/assets", StaticFiles(directory="frontend/assets"), name="assets")
+
+# Mount other static files (images, etc.)
+app.mount("/static", StaticFiles(directory="frontend"), name="static")
 
 # Load tokenizer and BERT model
 tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
@@ -81,4 +82,14 @@ def generate(text: str):
 
 @app.get("/")
 async def read_index():
+    return FileResponse("frontend/index.html")
+
+@app.get("/{file_path:path}")
+async def serve_static_files(file_path: str):
+    """Serve static files from frontend directory"""
+    import os
+    full_path = f"frontend/{file_path}"
+    if os.path.exists(full_path) and os.path.isfile(full_path):
+        return FileResponse(full_path)
+    # If file not found, serve index.html for React Router
     return FileResponse("frontend/index.html")