Upload 3 files

SPACES_README.md

@@ -4,7 +4,7 @@ emoji: 🔬
 colorFrom: purple
 colorTo: blue
 sdk: gradio
-sdk_version: 4.44.0
+sdk_version: 6.3.0
 app_file: app.py
 python_version: "3.10"
 pinned: false

app.py

@@ -4,6 +4,10 @@ HuggingFace Spaces entry point for diffviews.
 This file is the main entry point for HF Spaces deployment.
 It downloads required data and checkpoints on startup, then launches the Gradio app.
 
+Requirements:
+    Python 3.10+
+    Gradio 6.0+
+
 Environment variables:
     DIFFVIEWS_DATA_DIR: Override data directory (default: data)
     DIFFVIEWS_CHECKPOINT: Which checkpoint to download (dmd2, edm, all, none; default: dmd2)
@@ -13,30 +17,6 @@ Environment variables:
 import os
 from pathlib import Path
 
-# Monkey-patch gradio_client to handle additionalProperties: true
-# This fixes a bug where dict-typed State components cause schema errors
-# The bug is in _json_schema_to_python_type which recursively calls itself
-# with schema['additionalProperties'] which can be True (boolean)
-def _patch_gradio_client():
-    try:
-        import gradio_client.utils as client_utils
-
-        # Get the original function
-        original_func = client_utils._json_schema_to_python_type
-
-        def patched_json_schema_to_python_type(schema, defs=None):
-            # Handle case where schema is a boolean (additionalProperties: true)
-            if isinstance(schema, bool):
-                return "Any"
-            return original_func(schema, defs)
-
-        client_utils._json_schema_to_python_type = patched_json_schema_to_python_type
-        print("[Patch] Applied gradio_client _json_schema_to_python_type fix")
-    except Exception as e:
-        print(f"[Patch] Could not patch gradio_client: {e}")
-
-_patch_gradio_client()
-
 # Data source configuration
 DATA_REPO_ID = "mckell/diffviews_demo_data"
 CHECKPOINT_URLS = {
@@ -179,7 +159,13 @@ def main():
     ensure_data_ready(data_dir, checkpoints)
 
     # Import and launch visualizer
-    from diffviews.visualization.app import GradioVisualizer, create_gradio_app
+    import gradio as gr
+    from diffviews.visualization.app import (
+        GradioVisualizer,
+        create_gradio_app,
+        CUSTOM_CSS,
+        PLOTLY_HANDLER_JS,
+    )
 
     print("\nInitializing visualizer...")
     visualizer = GradioVisualizer(
@@ -196,7 +182,9 @@ def main():
         server_name="0.0.0.0",
         server_port=7860,
         share=False,  # Spaces handles public URL
-        show_api=False,  # Disable API docs to avoid schema generation bug
+        theme=gr.themes.Soft(),
+        css=CUSTOM_CSS,
+        js=PLOTLY_HANDLER_JS,
     )
 
 

requirements.txt

@@ -1,5 +1,5 @@
 # DiffViews - HuggingFace Spaces Requirements
-# Install diffviews package from GitHub
+# Requires Python 3.10+
 git+https://github.com/mckellcarter/diffviews.git
 
 # Core dependencies
@@ -14,14 +14,13 @@ tqdm>=4.60.0
 # Numba pinned for UMAP pickle compatibility
 numba==0.58.1
 
-# Visualization
-gradio==4.25.0
-gradio_client==0.15.0
+# Visualization - Gradio 6 for HF Spaces compatibility
+gradio>=6.0.0
 plotly>=5.18.0
 matplotlib>=3.5.0
 
-# HuggingFace Hub - pinned for Gradio 6 compatibility
-huggingface_hub>=0.19.0,<0.28.0
+# HuggingFace Hub
+huggingface_hub>=0.25.0
 
 # Optional but useful
 scipy>=1.7.0