import os
import gradio as gr
from dotenv import load_dotenv

load_dotenv()

# ------------------------------------------------------------
# Config
# ------------------------------------------------------------
HF_TOKEN = os.environ["HF_TOKEN"]   # optional, if you want to pass token for private space
SPACE_ID = "Fuzure/sheami"   # your Hugging Face Space


# ------------------------------------------------------------
# Protected Loader
# ------------------------------------------------------------
def build_app():
    with gr.Blocks() as demo:
        gr.Markdown("## 🔐 Sheami (Protected Access)")

        # Show login button (HF handles OAuth here)
        login_btn = gr.LoginButton()
        gr.Markdown("---")

        # Load the target Space inside this Blocks
        try:
            gr.load(
                SPACE_ID,
                src="spaces",
                hf_token=HF_TOKEN,
                alias="sheami",
            )
        except Exception as e:
            gr.Markdown(f"⚠️ Could not load Space: {e}")

    return demo


# ------------------------------------------------------------
# Main
# ------------------------------------------------------------
if __name__ == "__main__":
    app = build_app()
    app.launch()