Update app.py

app.py

@@ -32,7 +32,7 @@ from gradio_client import Client
 # ============================================================
 
 APP_NAME = "X-RUDRA"
-VERSION = "3.8.2"   # bumped
+VERSION = "3.8.3"   # bumped
 
 M1_REPO = os.getenv("M1_REPO", "Shrijanagain/M1")
 M2_REPO = os.getenv("M2_REPO", "Shrijanagain/M2")
@@ -75,10 +75,6 @@ def get_m2_client():
 
 # ============================================================
 # MODEL-BASED INTENT CLASSIFIER
-# ──────────────────────────────────────────────────────────────
-# Uses M1 (or M2) to decide ACTION or CASUAL.
-# No hardcoded keywords – the model decides.
-# If both fail, defaults to ACTION.
 # ============================================================
 
 CLASSIFIER_SYSTEM_PROMPT = """
@@ -99,8 +95,8 @@ def classify_intent(question: str) -> str:
         try:
             result = m1_client.predict(
                 prompt=prompt,
-                max_tokens=64,          # FIXED: was 10 → now 64 (minimum)
-                temperature=0.0,
+                max_tokens=64,
+                temperature=0.1,          # FIXED: was 0.0 → now 0.1
                 api_name="/generate"
             )
             if result:
@@ -122,8 +118,8 @@ def classify_intent(question: str) -> str:
         try:
             result = m2_client.predict(
                 prompt=prompt,
-                max_tokens=64,          # FIXED
-                temperature=0.0,
+                max_tokens=64,
+                temperature=0.1,          # FIXED
                 api_name="/generate"
             )
             if result:
@@ -139,9 +135,15 @@ def classify_intent(question: str) -> str:
         except Exception as e:
             print(f"[Classifier] M2 call failed: {e}")
 
-    # Fallback: default to ACTION (research) if both fail
-    print("[Classifier] Defaulting to ACTION")
-    return "ACTION"
+    # Fallback: if both fail, use simple heuristic (no keyword list)
+    # For very short messages (<=3 words), assume CASUAL to avoid unnecessary search.
+    word_count = len(question.split())
+    if word_count <= 3:
+        print("[Classifier] Fallback → CASUAL (short message)")
+        return "CASUAL"
+    else:
+        print("[Classifier] Fallback → ACTION (longer message)")
+        return "ACTION"
 
 
 # ============================================================
@@ -688,5 +690,4 @@ if __name__ == "__main__":
         print("HF_TOKEN set – rate limits reduced.")
     else:
         print("HF_TOKEN not set – you may experience rate limits. Set it as a Secret in your Space.")
-    demo.launch(server_name="0.0.0.0", server_port=PORT, css=CSS, show_error=True)
-    
+    demo.launch(server_name="0.0.0.0", server_port=PORT, css=CSS, show_error=True)