flashstudio 0.1.9__tar.gz → 0.2.0__tar.gz

{flashstudio-0.1.9 → flashstudio-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flashstudio
-Version: 0.1.9
+Version: 0.2.0
 Summary: Interactive Training & Inference UI for FlashDet — runs on Google Colab
 Author: FlashVision
 License: Apache-2.0

{flashstudio-0.1.9 → flashstudio-0.2.0}/README.md

@@ -287,33 +287,35 @@ FlashStudio/
 
 ## Troubleshooting
 
-### Colab: "localhost can't be reached"
+### Colab: "localhost can't be reached" or ngrok error
 
-This means the tunnel didn't connect. Try running in a new cell:
+`launch()` uses Colab's built-in proxy automatically. If it doesn't work, run this in a **new cell**:
 
 ```python
+import subprocess, sys, time
+
+proc = subprocess.Popen(
+    [sys.executable, "-m", "streamlit", "run",
+     __import__('flashstudio.launcher', fromlist=['_get_app_path'])._get_app_path(),
+     "--server.port", "8501", "--server.headless", "true",
+     "--server.enableCORS", "false", "--server.enableXsrfProtection", "false"],
+    stdout=subprocess.PIPE, stderr=subprocess.PIPE
+)
+time.sleep(5)
+
 from google.colab.output import eval_js
-print(eval_js('google.colab.kernel.proxyPort(8501)'))
+print(eval_js("google.colab.kernel.proxyPort(8501)"))
 ```
 
-Open the printed URL in a new tab.
+Open the printed URL in a new tab. No token or signup needed.
 
 ### Streamlit port already in use
 
 ```bash
-# Kill existing Streamlit processes
 pkill -f "streamlit run"
-
-# Then restart
 flashstudio --port 8501
 ```
 
-### `ModuleNotFoundError: No module named 'flashdet'`
-
-```bash
-pip install flashdet
-```
-
 ---
 
 ## Requirements

{flashstudio-0.1.9 → flashstudio-0.2.0}/flashstudio/launcher.py

@@ -55,14 +55,19 @@ def launch(port: int = 8501, share: bool = True, ngrok_token: str | None = None)
     app_path = _get_app_path()
 
     if is_colab():
-        _launch_colab(app_path, port, share, ngrok_token)
+        return _launch_colab(app_path, port, share, ngrok_token)
     else:
         _launch_local(app_path, port)
+        return None
 
 
 def _launch_colab(app_path: str, port: int, share: bool, ngrok_token: str | None):
     """Launch in Google Colab with automatic tunnel detection."""
     import time as _time
+    import urllib.request
+    import urllib.error
+
+    print("  Starting FlashStudio...", end="", flush=True)
 
     proc = subprocess.Popen(
         [sys.executable, "-m", "streamlit", "run", app_path,
@@ -74,28 +79,47 @@ def _launch_colab(app_path: str, port: int, share: bool, ngrok_token: str | None
         stderr=subprocess.PIPE,
     )
 
-    _time.sleep(3)
+    # Wait until Streamlit is actually responding (up to 30s)
+    ready = False
+    for i in range(30):
+        _time.sleep(1)
+        if proc.poll() is not None:
+            _, stderr = proc.communicate()
+            err_msg = stderr.decode("utf-8", errors="replace")[-500:] if stderr else "Unknown error"
+            print(f"\n\n  FlashStudio failed to start:\n  {err_msg}")
+            return proc
+        try:
+            urllib.request.urlopen(f"http://localhost:{port}/_stcore/health", timeout=2)
+            ready = True
+            break
+        except Exception:
+            print(".", end="", flush=True)
+
+    if ready:
+        print(" Ready!")
+    else:
+        print(" (still loading, URL may take a moment)")
 
     public_url = None
 
     # Method 1: Colab's built-in proxy (simplest, no install needed)
+    try:
+        from google.colab.output import eval_js
+        result = eval_js(f"google.colab.kernel.proxyPort({port})")
+        if result and str(result).startswith("http"):
+            public_url = str(result)
+    except Exception:
+        pass
+
+    # Method 2: ngrok (only if token is provided)
     if not public_url:
-        try:
-            from google.colab.output import eval_js
-            public_url = str(eval_js(f"google.colab.kernel.proxyPort({port})"))
-        except Exception:
-            pass
-
-    # Method 2: ngrok (if token provided or pyngrok available)
-    if not public_url and share:
         token = ngrok_token or os.environ.get("NGROK_TOKEN", "")
-        try:
-            from pyngrok import ngrok, conf
-            if token:
+        if token:
+            try:
+                from pyngrok import ngrok, conf
                 conf.get_default().auth_token = token
-            public_url = str(ngrok.connect(port, "http"))
-        except ImportError:
-            if token:
+                public_url = str(ngrok.connect(port, "http"))
+            except ImportError:
                 try:
                     subprocess.check_call(
                         [sys.executable, "-m", "pip", "install", "-q", "pyngrok"],
@@ -106,22 +130,8 @@ def _launch_colab(app_path: str, port: int, share: bool, ngrok_token: str | None
                     public_url = str(ngrok.connect(port, "http"))
                 except Exception:
                     pass
-        except Exception:
-            pass
-
-    # Method 3: localtunnel (no auth needed)
-    if not public_url and share:
-        try:
-            lt_proc = subprocess.run(
-                ["npx", "localtunnel", "--port", str(port)],
-                capture_output=True, text=True, timeout=15,
-            )
-            for line in lt_proc.stdout.splitlines():
-                if "https://" in line:
-                    public_url = line.strip().split()[-1]
-                    break
-        except Exception:
-            pass
+            except Exception:
+                pass
 
     print(f"\n{'=' * 60}")
     print("  FlashStudio is running!")
@@ -129,8 +139,8 @@ def _launch_colab(app_path: str, port: int, share: bool, ngrok_token: str | None
         print(f"  Open: {public_url}")
     else:
         print(f"  Port: {port}")
-        print("  ")
-        print("  If the URL doesn't work, try in a Colab cell:")
+        print("")
+        print("  Run this in a new cell to get the URL:")
         print(f"    from google.colab.output import eval_js")
         print(f"    print(eval_js('google.colab.kernel.proxyPort({port})'))")
     print(f"{'=' * 60}\n")
@@ -140,12 +150,27 @@ def _launch_colab(app_path: str, port: int, share: bool, ngrok_token: str | None
             from IPython.display import display, HTML
             display(HTML(
                 f'<a href="{public_url}" target="_blank" '
-                f'style="font-size:16px;padding:8px 16px;background:#7C3AED;'
-                f'color:white;border-radius:6px;text-decoration:none;">'
+                f'style="display:inline-block;font-size:16px;padding:10px 24px;'
+                f'background:#7C3AED;color:white;border-radius:8px;'
+                f'text-decoration:none;font-weight:600;margin:8px 0;">'
                 f'Open FlashStudio</a>'
             ))
         except Exception:
             pass
+    else:
+        try:
+            from IPython.display import display, HTML
+            display(HTML(
+                f'<div style="padding:12px;background:#FEF3C7;border-radius:8px;'
+                f'margin:8px 0;font-size:14px;">'
+                f'Run in a <b>new cell</b>:<br>'
+                f'<code style="background:#F3F4F6;padding:4px 8px;border-radius:4px;">'
+                f'from google.colab.output import eval_js; '
+                f'print(eval_js("google.colab.kernel.proxyPort({port})"))</code>'
+                f'</div>'
+            ))
+        except Exception:
+            pass
 
     return proc
 

{flashstudio-0.1.9 → flashstudio-0.2.0}/flashstudio.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flashstudio
-Version: 0.1.9
+Version: 0.2.0
 Summary: Interactive Training & Inference UI for FlashDet — runs on Google Colab
 Author: FlashVision
 License: Apache-2.0

{flashstudio-0.1.9 → flashstudio-0.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "flashstudio"
-version = "0.1.9"
+version = "0.2.0"
 description = "Interactive Training & Inference UI for FlashDet — runs on Google Colab"
 readme = "README_PYPI.md"
 license = {text = "Apache-2.0"}