Update app.py

app.py

@@ -418,7 +418,28 @@ def run_grammarllm(prompt, productions_json, regex_json):
     except Exception as e:
         return f"Errore durante l'inferenza: {str(e)}", None
 
-# Input di esempio per regex_json (stringa JSON)
+default_grammars = {
+    "Default Grammar": json.dumps({
+      "S*": ["<<positive>> A", "<<negative>> B", "<<neutral>> C"],
+      "A": ["<<happy>> D", "<<peaceful>> E", "<<joyful>> F"],
+      "B": ["<<sad>>", "<<angry>>", "<<frustrated>>"],
+      "C": ["<<calm>>", "<<indifferent>>", "<<unemotional>>"],
+      "D": ["<<enthusiastic>>"],
+      "E": ["<<content>>"],
+      "F": ["<<excited>>"]
+    }, indent=4),
+
+    "Other example": json.dumps({
+      'S*': ["<<(>> A B", "<<negligent>> V", '<<indifferent>>'],
+      'A': ["number", "letters", "ε"],
+      'B': ['<<)>> letters R'],
+      'R': ['C', 'D'], 
+      'C': ['<<calm>>', '<<indifferent>>', '<<unemotional>>'],
+      'D': ['<<angry>>', '<<frustrated>>'],
+      'V': ["<<option>>"],
+    }, indent=4),
+}
+
 default_regex_json = json.dumps({
   "regex_alfanum": "[a-zA-Z0-9]+",
   "regex_letters": "[a-zA-Z]+",
@@ -429,31 +450,77 @@ default_regex_json = json.dumps({
   "regex_(": "\\("
 }, indent=4)
 
-default_grammar_json = json.dumps({
-  "S*": ["<<positive>> A", "<<negative>> B", "<<neutral>> C"],
-  "A": ["<<happy>> D", "<<peaceful>> E", "<<joyful>> F"],
-  "B": ["<<sad>>", "<<angry>>", "<<frustrated>>"],
-  "C": ["<<calm>>", "<<indifferent>>", "<<unemotional>>"],
-  "D": ["<<enthusiastic>>"],
-  "E": ["<<content>>"],
-  "F": ["<<excited>>"]
-}, indent=4)
 
+def update_productions(grammar_choice):
+    # Aggiorna textbox productions al cambio preset
+    return default_grammars[grammar_choice]
+
+
+def load_file(file_obj):
+    if file_obj is None:
+        return ""
+    try:
+        content = file_obj.read().decode("utf-8")
+        # opzionale: validare JSON?
+        json.loads(content)
+        return content
+    except Exception as e:
+        return f"Errore nel caricamento file: {str(e)}"
+
+
+with gr.Blocks() as demo:
+
+    prompt_input = gr.Textbox(label="Inserisci prompt testuale")
+
+    with gr.Row():
+        grammar_choice = gr.Dropdown(
+            list(default_grammars.keys()),
+            label="Scegli Productions (JSON)",
+            value="Default Grammar",
+            interactive=True,
+            elem_id="grammar_choice"
+        )
+        productions_upload = gr.File(label="Carica file Productions (JSON)", file_types=['.json'])
+    
+    productions_text = gr.Textbox(label="Productions (JSON)", lines=10, value=default_grammars["Default Grammar"])
 
-demo = gr.Interface(
-    fn=run_grammarllm,
-    inputs=[
-            gr.Textbox(label="Inserisci prompt testuale"),
-            gr.Textbox(label="Inserisci productions (JSON)", lines=10, value=default_grammar_json),
-            gr.Textbox(label="Inserisci regex_dict (JSON)", lines=10, value=default_regex_json),
-    ],
-    outputs=[
-        gr.Textbox(label="Output generato"),
-        gr.File(label="Scarica ZIP"),
-    ],
-    title="GrammarLLM con output e download ZIP",
-    description="Inserisci prompt, productions e regex per generare testo e scaricare i file.",
-)
+    with gr.Row():
+        regex_upload = gr.File(label="Carica file Regex_dict (JSON)", file_types=['.json'])
+    
+    regex_text = gr.Textbox(label="Inserisci regex_dict (JSON)", lines=10, value=default_regex_json)
+
+    output_text = gr.Textbox(label="Output generato")
+    zip_file = gr.File(label="Scarica ZIP")
+
+    # Callback: quando cambio dropdown, aggiorno productions_text
+    grammar_choice.change(
+        fn=update_productions,
+        inputs=grammar_choice,
+        outputs=productions_text,
+    )
+
+    # Callback: quando carico file productions, aggiorno productions_text (override dropdown)
+    productions_upload.upload(
+        fn=load_file,
+        inputs=productions_upload,
+        outputs=productions_text,
+    )
+
+    # Callback: quando carico file regex, aggiorno regex_text
+    regex_upload.upload(
+        fn=load_file,
+        inputs=regex_upload,
+        outputs=regex_text,
+    )
+
+    # Al submit del form chiamo run_grammarllm
+    submit_btn = gr.Button("Genera output")
+
+    submit_btn.click(
+        fn=run_grammarllm,
+        inputs=[prompt_input, productions_text, regex_text],
+        outputs=[output_text, zip_file],
+    )
 
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    demo.launch()