Update app.py

app.py

@@ -1,4 +1,4 @@
- 
+
 import os
 import asyncio
 # Init with fake key
@@ -51,7 +51,8 @@ def oai_key_isvalid(api_key):
 load_dotenv()
 ss = st.session_state
 ss.prompt = None
- 
+if 'pending_prompt' not in st.session_state:
+    st.session_state.pending_prompt = None 
 # Set width of sidebar
 st.markdown(
     """
@@ -108,8 +109,7 @@ def run_prompt(prompt, file_path = '...', image_path = '...'):
         except openai.APIError:
             # Handle specific API errors here
             print("OpenAI API error, please try again!")
-    
-
+     
 pre_prompts = [
     'Generate a donor with PCE = 10% ',
     ('The history and development of Y6'
@@ -173,23 +173,18 @@ with st.sidebar:
     st.markdown('# What can I ask?')
     cols = st.columns(2)
     with cols[0]:
-        st.button(
-           r'👑  Generate a donor with PCE = 10%     🧨     ',
-            on_click=lambda: run_prompt(pre_prompts[0]),
-        )
-        st.button(
-            r'📚 The history and development of Y6     ',
-            on_click=lambda: run_prompt(pre_prompts[1]),
-        )
+    	if st.button(r'👑  Generate a donor with PCE = 10%     🧨     '):
+    	    st.session_state.pending_prompt = pre_prompts[0]
+    
+    	if st.button(r'📚 The history and development of Y6 '):
+    	    st.session_state.pending_prompt = pre_prompts[1]
+
     with cols[1]:
-        st.button(
-            r"🎄Predict the LogP of PM6  ",
-            on_click=lambda: run_prompt(pre_prompts[2]),
-        )
-        st.button(
-            r'💎 Predict the PCE of Y6',
-            on_click=lambda: run_prompt(pre_prompts[3]),
-        )
+    	if st.button(r"🎄Predict the LogP of PM6  "):
+        	     st.session_state.pending_prompt = pre_prompts[2]
+    
+    	if st.button(r'💎 Predict the PCE of Y6'):
+       	     st.session_state.pending_prompt = pre_prompts[3]
 
     st.selectbox(
             'Select the file type ',
@@ -219,15 +214,27 @@ with st.sidebar:
     st.markdown(f"# {len(tool_list)} available tools")
     st.dataframe(
         tool_list,
-        use_container_width=True,
+        width='stretch',
         hide_index=True,
         height=200
     )
 
+if st.session_state.pending_prompt is not None:
+    prompt_to_run = st.session_state.pending_prompt
+    st.session_state.pending_prompt = None 
+     
+    if not ss.get('model1_select') or not ss.get('model2_select'):
+        st.error("⚠️ Please input both model names in the sidebar first!")
+    else:
+        run_prompt(prompt_to_run)
+
 # Execute agent on user input
 if prompt := st.chat_input("Say something and/or attach files"):
-
-    if uploaded_file is not None:
+ 
+    if not ss.get('model1_select') or not ss.get('model2_select'):
+        st.error("⚠️ Please input both model names in the sidebar first!")
+    elif uploaded_file is not None:
+ 
         if ss.get('file_type') == 'CSV (.csv)':
             with tempfile.NamedTemporaryFile(  suffix ='.csv' ,delete=False) as f:
                  f.write(uploaded_file.read())