Update app.py

app.py

@@ -55,13 +55,27 @@ engine = agent_logic_module.RAG_Engine(
 print("--- [UI App] Bootstrap complete. Gradio UI is starting. ---")
 
 # ----------------------------------------------------------------------
-# 3. Core Gradio Chat Logic
+# 3. Import the usage‑counter helper (the file‑lock implementation)
+# ----------------------------------------------------------------------
+from usage_counter import increment_and_check
+
+# ----------------------------------------------------------------------
+# 4. Core Gradio Chat Logic – now wrapped with the quota check
 # ----------------------------------------------------------------------
 def respond(message: str, history: list[dict[str, str]]):
     """
     Called by Gradio for each user message.
-    Streams the response back to the UI.
+    If the quota is exhausted we immediately return an error message.
+    Otherwise we stream the response from the RAG engine.
     """
+    # ---- quota check (atomic) ----
+    if not increment_and_check():
+        # The limit has been reached – we do **not** call the engine.
+        # Yield a single message so the UI shows it instantly.
+        yield "❌ **Quota exceeded** – the maximum number of allowed requests has been reached."
+        return
+
+    # ---- normal RAG processing ----
     final_response = engine.get_response(message, history)
 
     # Simple "typing" effect – yield partial strings
@@ -72,9 +86,8 @@ def respond(message: str, history: list[dict[str, str]]):
         yield response
 
 # ----------------------------------------------------------------------
-# 4. UI Layout – Tips + Chat + Footer
+# 5. UI Layout – Tips + Chat + Footer (unchanged)
 # ----------------------------------------------------------------------
-# 4.1  Tips (you can edit this markdown as you wish)
 tips_md = r"""
 ## 📋 Tips menggunakan **PRECISE RAG Agent**
 
@@ -91,7 +104,6 @@ tips_md = r"""
 
 """
 
-# 4.2  Footer – the old description / notes
 footer_md = r"""
 ---  
 **Frameworks**: LangChain + FAISS  
@@ -100,7 +112,6 @@ footer_md = r"""
 
 """
 
-# 4.3  Chat component (no description here)
 chatbot = gr.ChatInterface(
     respond,
     type="messages",
@@ -114,22 +125,7 @@ chatbot = gr.ChatInterface(
     theme=gr.themes.Soft(),
 )
 
-# 4.4  Assemble everything inside a Blocks container
 with gr.Blocks() as demo:
-    # Optional: add a small vertical space at the top 
-    chatbot.render()             # <-- Main chat UI
-    gr.Markdown(tips_md)          # <-- Tips section (renders LaTeX)
-    gr.Markdown(footer_md)        # <-- Footer (old description)
-
-# ----------------------------------------------------------------------
-# 5. Launch
-# ----------------------------------------------------------------------
-if __name__ == "__main__":
-    allowed_user = _secret("CHAT_USER")
-    allowed_pass = _secret("CHAT_PASS")
-    demo.launch(
-        auth=(allowed_user, allowed_pass),
-        server_name="0.0.0.0",
-        ssr_mode=False,
-        server_port=7860,
-    )
+    chatbot.render()
+    gr.Markdown(tips_md)
+    gr.Markdown(footer_md)