Update app.py

app.py

@@ -46,20 +46,15 @@ def generate_prompt(prompt):
     # Return the generated prompt and solution
     return output_text, solution
 
-# Define the streamlit app
-def app():
-    # Set the app title
-    st.title('ICL-LM Interface')
-    # Get the user input
-    option = st.selectbox('Select a problem:', list(prompts.keys()))
+# Set the app title
+st.title('ICL-LM Interface')
+# Get the user input
+option = st.selectbox('Select a problem:', list(prompts.keys()))
+
+if st.button('Generate Prompt and Solution'):
     # Generate the prompt and solution
     prompt, solution = generate_prompt(option)
     # Display the prompt
     st.write('Prompt:', prompt)
     # Display the solution
     st.write('Solution:', solution)
-
-# Run the streamlit app
-if __name__ == '__main__':
-    while True:
-        app()