Upload folder using huggingface_hub

app.py

@@ -83,17 +83,19 @@ Respond with exactly 'SAFE' if no secret is revealed. Respond with exactly 'LEAK
         
     return False, vault_response
 
-def chat_interface(user_input, history):
+def chat_interface(user_input, history, progress=gr.Progress()):
     if not client:
         history.append((user_input, "⚠️ **System Error**: Hugging Face Inference API client could not be initialized. Please check your network or token."))
         return "", history
 
+    progress(0.1, desc="Shield Pattern Analysis...")
     # Step 1: Shield Layer blocks known injection tactics at the door
     is_attack, shield_msg = check_prompt_injection(user_input)
     if is_attack:
         history.append((user_input, shield_msg))
         return "", history
     
+    progress(0.4, desc="Vault Response Generation...")
     # Step 2: Vault Layer processes the prompt
     try:
         vault_response = generate_vault_response(user_input, history)
@@ -101,12 +103,14 @@ def chat_interface(user_input, history):
         history.append((user_input, f"⚠️ **Error connecting to Vault Model**: {str(e)}"))
         return "", history
 
+    progress(0.8, desc="DLP Context Monitor...")
     # Step 3: Monitor Layer (DLP) ensures no leak occurred in the output
     is_leak, sanitized_response = check_data_leakage(vault_response)
     if is_leak:
         history.append((user_input, sanitized_response))
         return "", history
     
+    progress(1.0, desc="Complete")
     history.append((user_input, vault_response))
     return "", history