Upload app.py with huggingface_hub

app.py

@@ -311,6 +311,13 @@ def _is_zerogpu_queue_timeout(err: Exception) -> bool:
     return ZEROGPU_QUEUE_HINT in str(err)
 
 
+def _spaces_user_logged_in(request: gr.Request | None) -> bool:
+    """True when a Spaces OAuth user is attached to this request."""
+    if request is None:
+        return False
+    return bool(getattr(request, "username", None))
+
+
 
 def _use_hf_model() -> bool:
     """Check if we should use the fine-tuned HF model."""
@@ -321,10 +328,20 @@ def _use_hf_model() -> bool:
     return os.getenv("VERIS_USE_HF", "").lower() in ("1", "true", "yes")
 
 
-def classify(description: str, api_key: str) -> str:
+def classify(description: str, api_key: str, request: gr.Request | None = None) -> str:
     """Classify an incident — uses HF model on Spaces, OpenAI otherwise."""
     if not description.strip():
         return "Please enter an incident description."
+    if IS_SPACES and not _spaces_user_logged_in(request):
+        return json.dumps(
+            {
+                "error": (
+                    "Please sign in with Hugging Face in this app before running inference. "
+                    "ZeroGPU quota is per logged-in user."
+                )
+            },
+            indent=2,
+        )
 
     use_hf = _use_hf_model()
 
@@ -364,10 +381,15 @@ def classify(description: str, api_key: str) -> str:
         return json.dumps({"error": str(e)}, indent=2)
 
 
-def ask(question: str, api_key: str) -> str:
+def ask(question: str, api_key: str, request: gr.Request | None = None) -> str:
     """Answer a VERIS question — uses HF model on Spaces, OpenAI otherwise."""
     if not question.strip():
         return "*Please enter a question.*"
+    if IS_SPACES and not _spaces_user_logged_in(request):
+        return (
+            "**Error:** Please sign in with Hugging Face in this app before running inference. "
+            "ZeroGPU quota is per logged-in user."
+        )
 
     use_hf = _use_hf_model()
 
@@ -547,6 +569,12 @@ def build_app() -> gr.Blocks:
                 </span>
             </div>
             """)
+            with gr.Row():
+                gr.Markdown(
+                    "**Required:** Click Sign in with Hugging Face below so ZeroGPU usage "
+                    "counts against your account quota."
+                )
+                gr.LoginButton("Sign in with Hugging Face")
         else:
             gr.HTML("""
             <div class="model-banner">