update for model comparison, add helpful interface labels

app.py

@@ -1,12 +1,40 @@
+from typing import Any
 import gradio as gr
 from listener import Listener
 
-listener = Listener("pragmatic-programs/pragmatic-ft-listener", {"do_sample": True, "num_return_sequences": 100, "num_beams": 1, "temperature": 1, "top_p": 0.9})
+pragmatic_listener = Listener("pragmatic-programs/pragmatic-ft-listener", {"do_sample": True, "num_return_sequences": 100, "num_beams": 1, "temperature": 1, "top_p": 0.9, "max_new_tokens": 128})
+literal_listener = Listener("pragmatic-programs/listener-suffix-idx-300k", {"do_sample": True, "num_return_sequences": 100, "num_beams": 1, "temperature": 1, "top_p": 0.9, "max_new_tokens": 128})
 
 def synthesize(context):
-    spec = [[[s[:-1], s[-1]] for s in context.split(' ')]]
-    return listener.synthesize(spec).programs[0][0]
+    if len(context.strip()) == 0:
+        return "Empty specification", "Empty specification"
 
+    spec = [[[s[:-1], s[-1]] for s in context.strip().split(' ')]]
+    if not all([len(s) > 0 and l in ['+', '-'] for s, l in spec[0]]):
+        return "Invalid specification", "Invalid specification"
 
-iface = gr.Interface(fn=synthesize, inputs="text", outputs="text")
+    pragmatic_outputs = pragmatic_listener.synthesize(spec).programs
+    literal_outputs = literal_listener.synthesize(spec).programs
+
+    if len(pragmatic_outputs[0]) > 0:
+        pragmatic_program = pragmatic_outputs[0][0]
+    else:
+        pragmatic_program = "No program found"
+    
+    if len(literal_outputs[0]) > 0:
+        literal_program = literal_outputs[0][0]
+    else:
+        literal_program = "No program found"
+
+    return pragmatic_program, literal_program
+
+
+iface = gr.Interface(
+    fn=synthesize, 
+    inputs=gr.Textbox(lines=1, label="Examples", info="Enter a list of examples, separated by spaces. Each example is the string followed by a + or - indicating whether it should be accepted or rejected by the synthesized regex."),
+    outputs=[gr.Textbox(lines=1, label="Pragmatic model"), gr.Textbox(lines=1, label="Literal model")],
+    examples=["ab+ aabb+ abb+ abab-", "b0b+ aa0000bb+"],
+    title="Synthesizing regular expressions from examples",
+
+    )
 iface.launch()