Update app2.py

app2.py

@@ -10,7 +10,7 @@ else:
     print("[SYSTEM] Cache is clean.")
 
 # --- YOUR ORIGINAL CODE STARTS HERE ---
-from flask import Flask, request, jsonify
+from flask import Flask, request, jsonify, send_from_directory # Added send_from_directory
 from flask_sock import Sock
 from transformers import AutoModel
 import torch
@@ -36,6 +36,11 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model = model.to(device)
 print(f"[SYSTEM] Model loaded successfully onto device: {device}")
 
+@app.route('/')
+def serve_ui():
+    # This tells Flask to send the index.html file to the user's browser
+    return send_from_directory('.', 'index.html')
+
 # --- THE NEW WEBSOCKET PIPELINE ---
 @sock.route('/api/generate_stream')
 def stream_motion(ws):