Update app.py

app.py

@@ -1,32 +1,27 @@
 import gradio as gr
 from huggingface_hub import HfApi, login, upload_file
 
-# Local storage for rules
 rules = {}
 
-# Load rules from a file at startup
 def load_rules():
-    # Download rules file from HF dataset/repo (use huggingface_hub or requests)
     global rules
     try:
         with open("rules.txt", "r") as f:
             lines = f.readlines()
-            rules = dict(line.strip().split("=>") for line in lines if "=>" in line)
-    except Exception as e:
+            rules = dict(line.strip().split("=>", 1) for line in lines if "=>" in line)
+    except Exception:
         rules = {}
 
 def chatbot_response(message, history):
-    # Match rule, else generic fallback
+    # message is a string, history is now a list of dicts in "messages" format
     for k, v in rules.items():
         if k in message:
             return v
     return "No matching rule. Submit an example!"
 
 def submit_rule(user_input, rule_response, write_token):
-    # Add new rule
     global rules
     rules[user_input] = rule_response
-    # Save locally
     with open("rules.txt", "a") as f:
         f.write(f"{user_input}=>{rule_response}\n")
     # Upload using user's write token
@@ -45,15 +40,19 @@ def submit_rule(user_input, rule_response, write_token):
 load_rules()
 
 with gr.Blocks() as demo:
-    chat = gr.ChatInterface(chatbot_response)
+    chat = gr.ChatInterface(
+        fn=chatbot_response,
+        type="messages"  # Use the recommended format
+    )
     rule_input = gr.Textbox(label="User Example")
     rule_output = gr.Textbox(label="Response")
     token_input = gr.Textbox(label="Write Token", type="password")
     submit = gr.Button("Submit Rule")
+    status_output = gr.Textbox(label="Status", interactive=False)
     submit.click(
         submit_rule,
         inputs=[rule_input, rule_output, token_input],
-        outputs=gr.Textbox(label="Status")
+        outputs=status_output
     )
 
 demo.launch()