checking SPACE_HOST to add https

main.py

@@ -11,6 +11,7 @@ import pandas as pd
 from fastapi import FastAPI
 from fastapi.responses import HTMLResponse, RedirectResponse
 from fastapi.middleware.cors import CORSMiddleware
+from os import getenv
 from sql import Q
 
 from itables import options as itoptions, to_html_datatable
@@ -102,7 +103,12 @@ def query_from_request(sql_input, definitions, request: gr.Request):
         q.save()
         query_id = q.id
         alias = q.alias
-        iframe_src = f"/q?query_id={query_id}"
+        hf_space_host = getenv("SPACE_HOST")
+        if hf_space_host:
+            iframe_src = f"https://{hf_space_host}/q?query_id={query_id}"
+        else:
+            iframe_src = f"/q?query_id={query_id}"
+
 
     result = f"""
     <div id="resultContainer">