update app.py

app.py

@@ -69,10 +69,19 @@ def user_submit(user_message: str, history: List[Tuple[str, str]]):
     return "", history
 
 
+def _oauth_or_env_token(hf_token: gr.OAuthToken | None):
+    try:
+        if hf_token and getattr(hf_token, "token", None):
+            return hf_token.token
+    except Exception:
+        pass
+    return os.getenv("HF_TOKEN")
+
+
 def bot_reply(history: List[Tuple[str, str]], temperature: float = 0.7, top_p: float = 0.95, max_tokens: int = 512, hf_token: gr.OAuthToken = None):
     last_user = history[-1][0] if history else ""
     messages = _build_messages(history[:-1], last_user)
-    client = InferenceClient(token=(hf_token.token if hf_token else None), model=HF_MODEL)
+    client = InferenceClient(token=_oauth_or_env_token(hf_token), model=HF_MODEL)
 
     acc = ""
     for event in client.chat_completion(
@@ -99,7 +108,7 @@ def summarize(history: List[Tuple[str, str]], hf_token: gr.OAuthToken = None):
     if not transcript.strip():
         return {"note": "No conversation yet. Ask Maya some questions first."}
 
-    client = InferenceClient(token=(hf_token.token if hf_token else None), model=HF_MODEL)
+    client = InferenceClient(token=_oauth_or_env_token(hf_token), model=HF_MODEL)
     messages = [
         {"role": "system", "content": SUMMARY_SYSTEM_PROMPT},
         {
@@ -126,9 +135,13 @@ def summarize(history: List[Tuple[str, str]], hf_token: gr.OAuthToken = None):
 
 
 with gr.Blocks(title="Maya – Klinik Owner (Role-Play)") as demo:
-    # Top login section (compatible with Gradio 4.x)
-    gr.Markdown("## Sign in to use the model")
-    gr.LoginButton()
+    # Top login section: enable OAuth on Spaces, skip locally
+    is_space = bool(os.getenv("SPACE_ID") or os.getenv("SYSTEM") == "spaces")
+    if is_space:
+        gr.Markdown("## Sign in to use the model")
+        gr.LoginButton()
+    else:
+        gr.Markdown("Tip: set env var `HF_TOKEN` for local use.")
     gr.Markdown("Model: `" + HF_MODEL + "`")
 
     gr.Markdown(