Update app.py

app.py

@@ -22,52 +22,55 @@ example_assemblies = {
 
 # Streamlit App
 st.title("The Assembly Equation Interactive Simulator")
-st.write("""
-The Assembly Equation is defined as:
-\( A = \sum_{i=1}^{N} e^{a_i} \left(\frac{n_i - 1}{N_T}\right) \)
 
-Where:
-- \( A \) is the assembly of the ensemble
-- \( a_i \) is the assembly index of object \( i \)
-- \( n_i \) is the copy number of object \( i \)
-- \( N \) is the total number of unique objects
-- \( N_T \) is the total number of objects in the ensemble
-""")
+# Display the Assembly Equation using LaTeX
+st.latex(r'''
+A = \sum_{i=1}^{N} e^{a_i} \left(\frac{n_i - 1}{N_T}\right)
+''')
+
+st.write("Where:")
+st.latex(r'''
+A \quad \text{is the assembly of the ensemble}
+''')
+st.latex(r'''
+a_i \quad \text{is the assembly index of object } i
+''')
+st.latex(r'''
+n_i \quad \text{is the copy number of object } i
+''')
+st.latex(r'''
+N \quad \text{is the total number of unique objects}
+''')
+st.latex(r'''
+N_T \quad \text{is the total number of objects in the ensemble}
+''')
 
 # Sidebar for inputs
 st.sidebar.header("Input Parameters")
-N = st.sidebar.number_input("Enter the total number of unique objects (N):", min_value=1, value=len(example_assemblies))
 N_T = st.sidebar.number_input("Enter the total number of objects in the ensemble (N_T):", min_value=1, value=50)
 
-# Inputs for each object
-ai_values = []
-ni_values = []
-objects = list(example_assemblies.keys())
-
-for i in range(N):
-    obj = objects[i]
-    ai = st.sidebar.number_input(f"Enter the assembly index of {obj} (a{i+1}):", value=example_assemblies[obj]["ai"], key=f"a{i+1}")
-    ni = st.sidebar.number_input(f"Enter the copy number of {obj} (n{i+1}):", min_value=0, value=example_assemblies[obj]["ni"], key=f"n{i+1}")
-    ai_values.append(ai)
-    ni_values.append(ni)
+# Display example assemblies
+st.sidebar.write("Example Assemblies:")
+for assembly, params in example_assemblies.items():
+    st.sidebar.write(f"{assembly}: ai = {params['ai']}, ni = {params['ni']}")
 
 # Button to calculate assembly
 if st.sidebar.button('Calculate Assembly'):
+    ai_values = [params['ai'] for params in example_assemblies.values()]
+    ni_values = [params['ni'] for params in example_assemblies.values()]
     A = calculate_assembly(ai_values, ni_values, N_T)
     st.write("The assembly of the ensemble (A) is:", A)
     
     # Plotting the assembly index
     fig, ax = plt.subplots()
-    ax.bar(objects, ai_values, color='skyblue')
+    ax.bar(example_assemblies.keys(), ai_values, color='skyblue')
     ax.set_xlabel('Objects')
     ax.set_ylabel('Assembly Index')
     ax.set_title('Assembly Index of Each Object')
     st.pyplot(fig)
 else:
-    st.write("Enter values and press 'Calculate Assembly' to see the result.")
+    st.write("Press 'Calculate Assembly' to see the result.")
 
 # Showing the dictionary of examples
 st.write("Example inputs for simple assemblies:")
 st.json(example_assemblies)
-
-# Run the app with streamlit run <this_script_name>.py