Update demo.py

demo.py

@@ -5,47 +5,7 @@ import os
 import re
 from gradio_client import Client, file
 
-# --- Client Setup ---
-# Try connecting using environment variables first
-client_url = "http://127.0.0.1:7860" # Default if no env var
-try:
-    # Use 'src' if defined (common for HF Spaces)
-    if 'src' in os.environ:
-        client_url = os.environ['src']
-        print(f"Connecting to client URL (src): {client_url}")
-        client = Client(client_url)
-    # Fallback to host/key if src isn't defined but host/key are
-    elif 'host' in os.environ and 'key' in os.environ:
-        client_url = os.environ['host']
-        api_key = os.environ['key']
-        print(f"Connecting to client URL (host): {client_url} using API key.")
-        # Note: gradio_client might expect hf_token for private spaces
-        client = Client(client_url, hf_token=api_key) # Assuming key is hf_token
-    # Fallback to just host if only host is defined
-    elif 'host' in os.environ:
-         client_url = os.environ['host']
-         print(f"Connecting to client URL (host): {client_url} (public/no key)")
-         client = Client(client_url)
-    # Fallback to the hardcoded default if no relevant env vars found
-    else:
-        print(f"No suitable environment variables (src, host/key, host) found.")
-        print(f"Attempting connection to default URL: {client_url}")
-        client = Client(client_url) # Use the default
-
-    print("Gradio Client connected successfully.")
-    # Optional: Check API endpoints
-    # print(client.view_api(print_info=True))
-except Exception as e:
-    print(f"Error connecting Gradio Client to {client_url}: {e}")
-    print("Please ensure the source Gradio app is running and the URL/credentials are correct.")
-    # Provide a dummy client to prevent crashes
-    class DummyClient:
-        def predict(*args, **kwargs):
-            print("!!! Gradio Client not connected. Prediction will fail. !!!")
-            import numpy as np
-            return (44100, np.zeros(1)) # Sample rate, empty numpy array
-    client = DummyClient()
-
+client = Client(os.environ['src'])
 
 # --- UI Data Loading (Client-Side) ---
 BASE_PATH = "Inference"