Update app.py

app.py

@@ -1,108 +1,56 @@
 import os
-import subprocess
-import threading
-import time
+import sys
 
-import gradio as gr
-
-
-STUDIO_HOST = "127.0.0.1"
-STUDIO_PORT = 8888
+# ---------------------------------------------------------
+# Unsloth Studio environment
+# ---------------------------------------------------------
 
-# Keep Studio data inside the Space
 os.environ.setdefault(
     "UNSLOTH_STUDIO_HOME",
     "/tmp/unsloth_studio",
 )
 
+os.makedirs(
+    os.environ["UNSLOTH_STUDIO_HOME"],
+    exist_ok=True,
+)
 
-def start_studio():
-    print("🦥 Starting Unsloth Studio...")
-
-    os.makedirs(os.environ["UNSLOTH_STUDIO_HOME"], exist_ok=True)
-
-    cmd = [
-        "unsloth",
-        "studio",
-        "-H",
-        STUDIO_HOST,
-        "-p",
-        str(STUDIO_PORT),
-    ]
-
-    process = subprocess.Popen(
-        cmd,
-        stdout=subprocess.PIPE,
-        stderr=subprocess.STDOUT,
-        text=True,
-        bufsize=1,
-    )
-
-    def log_output():
-        for line in process.stdout:
-            print("[Unsloth Studio]", line.rstrip())
-
-    threading.Thread(
-        target=log_output,
-        daemon=True,
-    ).start()
-
-    return process
-
-
-studio_process = start_studio()
-
-
-def wait_for_studio():
-    import urllib.request
-
-    for _ in range(60):
-        try:
-            urllib.request.urlopen(
-                f"http://{STUDIO_HOST}:{STUDIO_PORT}",
-                timeout=1,
-            )
-            return True
-        except Exception:
-            time.sleep(1)
+# ---------------------------------------------------------
+# Import the REAL Unsloth Studio FastAPI application
+# ---------------------------------------------------------
 
-    return False
+from studio.backend.main import app as studio_app
 
+# ---------------------------------------------------------
+# Gradio
+# ---------------------------------------------------------
 
-if not wait_for_studio():
-    raise RuntimeError(
-        "Unsloth Studio failed to start."
-    )
+import gradio as gr
 
 
-# Gradio Space wrapper
 with gr.Blocks(title="Unsloth Studio") as demo:
 
     gr.Markdown(
         """
         # 🦥 Unsloth Studio
 
-        Unsloth Studio is running inside this Hugging Face Space.
+        Full Unsloth Studio running inside a Hugging Face Space.
         """
     )
 
-    gr.HTML(
-        f"""
-        <iframe
-            src="http://{STUDIO_HOST}:{STUDIO_PORT}"
-            style="
-                width: 100%;
-                height: 900px;
-                border: none;
-                border-radius: 10px;
-            "
-        ></iframe>
-        """
-    )
+
+# ---------------------------------------------------------
+# Mount Studio into Gradio
+# ---------------------------------------------------------
+
+demo.app = studio_app
 
 
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
+    import uvicorn
+
+    uvicorn.run(
+        studio_app,
+        host="0.0.0.0",
+        port=7860,
     )