added spin

app.py

@@ -341,14 +341,15 @@ def main():
     st.title("$\\mathscr{L}$agrangian Generator")
     st.markdown(" ### For a set of chosen fields, this model generates the corresponding Lagrangian which encodes all interactions and dynamics of the fields.")
     
-    st.markdown(" #### This is a simple demo of our smaller [BART](https://arxiv.org/abs/1910.13461)-based model with 110M parameters")
+    st.markdown(" #### This is a demo of our [BART](https://arxiv.org/abs/1910.13461)-based model with ca 360M parameters")
 
-    st.markdown(" ##### :violet[Due to computational resources, we limit the number of fields to 2 and spin = 0]")
-    st.markdown(" ##### Choose up to two different fields:")
+    st.markdown(" ##### :violet[Due to computational resources, we limit the number of fields to 3 and the maximum length of the generated Lagrangian to 512 tokens.]")
+    st.markdown(" ##### Choose up to three different fields:")
 
     su2_options = ["$1$", "$2$", "$3$"]
     su3_options = ["$1$", "$3$", "$\\bar{3}$"]
     u1_options = ["-1","-2/3", "-1/2", "-1/3", "0","1/3" ,"1/2", "2/3", "1"]
+    spin_options = ["1", "1/2"]
     
     # Initialize or update session state variables
     if 'count' not in st.session_state:
@@ -357,17 +358,17 @@ def main():
         st.session_state.field_strings = []  # Stores the generated field strings
 
     with st.form("field_selection"):
+        spin_selection = st.radio("Select spin value:", spin_options)
         su2_selection = st.radio("Select $\\mathrm{SU}(2)$ value:", su2_options)
         su3_selection = st.radio("Select $\\mathrm{SU}(3)$ value:", su3_options)
         u1_selection = st.radio("Select $\\mathrm{U}(1)$ value:", u1_options)
         submitted = st.form_submit_button("Add field")
         if submitted:
             if st.session_state.count < 3:
-                sp_value = 0  # Assume SP is always 0
-                field_string = generate_field(sp_value, su2_selection, su3_selection, u1_selection)
+                field_string = generate_field(spin_selection, su2_selection, su3_selection, u1_selection)
                 st.session_state.field_strings.append(field_string)  # Save generated field string
                 st.session_state.count += 1  # Increment button press count
-            elif st.session_state.count >= 2:
+            elif st.session_state.count >= 3:
                 st.write("You have reached the maximum number of fields we allow in this demo.")
     clear_fields = st.button("Clear fields")
     if clear_fields: