Update app.py

Added history handling. Set a hard upper-limit on history. Handled inputs properly. now, it should work.

app.py

@@ -12,33 +12,6 @@ import gradio as gr
 
 set_random_seed(65536)
 
-
-def respond(
-    message,
-        message1,
-        message2
-):
-    print(message)
-    print(message1)
-    print(message2)
-    return is_malicious_sql(message, 0.5)
-
-
-
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Slider(minimum=0.01, maximum=0.99, value=0.75, step=0.01, label="Detection Probability Threshold "),
-    ],
-)
-
-
-if __name__ == "__main__":
-    demo.launch()
-
 pandarallel.initialize(use_memory_fs=True, nb_workers=cpu_count())
 model = load_model('./sqid.keras')
 
@@ -125,3 +98,35 @@ def is_malicious_sql(sql, threshold):
     if preds > float(threshold):
         return 'Malicious'
     return 'Safe'
+
+
+def respond(
+    message,
+    history,
+        threshold
+):
+    if len(history) > 5:
+        history = history[1:]
+    for val in history:
+        if val[0].lower().strip() == message.lower().strip():
+            return val[1]
+
+    val = (message.lower().strip(), is_malicious_sql(message, threshold))
+    history.append(val)
+    return val[1]
+
+
+
+"""
+For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
+"""
+demo = gr.ChatInterface(
+    respond,
+    additional_inputs=[
+        gr.Slider(minimum=0.01, maximum=0.99, value=0.75, step=0.01, label="Detection Probability Threshold "),
+    ],
+)
+
+
+if __name__ == "__main__":
+    demo.launch()