use host url

app.py

@@ -21,7 +21,9 @@ NUM_USER_DATASETS = 10
 SESSIONS_DIR = "s"
 SESSION_DIR_SPACES_SYMLINK = os.path.join("/data", SESSIONS_DIR)
 SPACE_ID = os.getenv("SPACE_ID")
-URL = f"https://huggingface.co/spaces/{SPACE_ID}" if SPACE_ID else "http://127.0.0.1:7870"
+SPACE_HOST = os.getenv("SPACE_HOST")
+APP_URL = f"https://huggingface.co/spaces/{SPACE_ID}" if SPACE_ID else "http://127.0.0.1:7870"
+HOST_URL = f"https://{SPACE_HOST}" if SPACE_HOST else "http://127.0.0.1:7870"
 css = """
 .transparent-dropdown, .transparent-dropdown .container .wrap, .transparent-accordion  {
     background: var(--body-background-fill);
@@ -130,10 +132,10 @@ with gr.Blocks(css=css) as demo:
             split_dropdown: gr.Dropdown(**splits),
             session_state: session,
             dataframe: gr.DataFrame(**input_dataframe),
-            share_link_textbox: f"{URL}?{SESSIONS_DIR}={session}",
+            share_link_textbox: f"{APP_URL}?{SESSIONS_DIR}={session}",
             use_locally_markdown: (
-                f"""In DuckDB:\n\n```sql\nATTACH '{URL}/gradio_api/file={SESSIONS_DIR}/{session}.duckdb AS db';\nUSE db;\nSELECT * FROM edited_dataset LIMIT 5;\n```\n\n"""
-                f"""In Python:\n\n```python\nimport duckdb\n\nduckdb.sql("ATTACH '{URL}/gradio_api/file={SESSIONS_DIR}/{session}.duckdb' AS db")\nduckdb.sql("USE db")\ndf = duckdb.sql("SELECT * FROM edited_dataset LIMIT 5").df()\n```"""
+                f"""In DuckDB:\n\n```sql\nATTACH '{HOST_URL}/gradio_api/file={SESSIONS_DIR}/{session}.duckdb AS db';\nUSE db;\nSELECT * FROM edited_dataset LIMIT 5;\n```\n\n"""
+                f"""In Python:\n\n```python\nimport duckdb\n\nduckdb.sql("ATTACH '{HOST_URL}/gradio_api/file={SESSIONS_DIR}/{session}.duckdb' AS db")\nduckdb.sql("USE db")\ndf = duckdb.sql("SELECT * FROM edited_dataset LIMIT 5").df()\n```"""
             )
         }