Update app.py

app.py

@@ -55,27 +55,13 @@ engine = agent_logic_module.RAG_Engine(
 print("--- [UI App] Bootstrap complete. Gradio UI is starting. ---")
 
 # ----------------------------------------------------------------------
-# 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
+# 3. Core Gradio Chat Logic
 # ----------------------------------------------------------------------
 def respond(message: str, history: list[dict[str, str]]):
     """
     Called by Gradio for each user message.
-    If the quota is exhausted we immediately return an error message.
-    Otherwise we stream the response from the RAG engine.
+    Streams the response back to the UI.
     """
-    # ---- 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
@@ -86,8 +72,9 @@ def respond(message: str, history: list[dict[str, str]]):
         yield response
 
 # ----------------------------------------------------------------------
-# 5. UI Layout – Tips + Chat + Footer (unchanged)
+# 4. UI Layout – Tips + Chat + Footer
 # ----------------------------------------------------------------------
+# 4.1  Tips (you can edit this markdown as you wish)
 tips_md = r"""
 ## 📋 Tips menggunakan **PRECISE RAG Agent**
 
@@ -104,6 +91,7 @@ tips_md = r"""
 
 """
 
+# 4.2  Footer – the old description / notes
 footer_md = r"""
 ---  
 **Frameworks**: LangChain + FAISS  
@@ -112,6 +100,7 @@ footer_md = r"""
 
 """
 
+# 4.3  Chat component (no description here)
 chatbot = gr.ChatInterface(
     respond,
     type="messages",
@@ -125,7 +114,22 @@ chatbot = gr.ChatInterface(
     theme=gr.themes.Soft(),
 )
 
+# 4.4  Assemble everything inside a Blocks container
 with gr.Blocks() as demo:
-    chatbot.render()
-    gr.Markdown(tips_md)
-    gr.Markdown(footer_md)
+    # 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,
+    )