Update app.py

app.py

@@ -98,6 +98,7 @@ function_map = {
 # --- Conversation memory: Use Streamlit session state
 if "messages" not in st.session_state:
     st.session_state.messages = []
+
 if "temp_input" not in st.session_state:
     st.session_state.temp_input = ""
 
@@ -135,78 +136,74 @@ for i, msg in enumerate(st.session_state.messages[1:]):  # Skip system message f
         except Exception:
             st.markdown(f"<b>Function '{msg['name']}' output:</b> {msg['content']}", unsafe_allow_html=True)
 
-# --- User input box at bottom (like ChatGPT)
-if df is not None:
-    user_input = st.text_input("Your message:", value=st.session_state.temp_input, key="temp_input")
-    send = st.button("Send")
-else:
-    user_input = None
-    send = False
-
-if send and user_input and user_input.strip():
-    # Append user's message to conversation
-    st.session_state.messages.append({"role": "user", "content": user_input})
-
-    # Compose messages for OpenAI (entire chat history)
-    chat_messages = st.session_state.messages.copy()
-
-    # First OpenAI call: Check for function call
-    chat_resp = requests.post(
-        "https://api.openai.com/v1/chat/completions",
-        headers=HEADERS,
-        json={
-            "model": "gpt-3.5-turbo-1106",
-            "messages": chat_messages,
-            "functions": function_schema,
-            "function_call": "auto",
-            "temperature": 0,
-            "max_tokens": 1000,
-        },
-        timeout=60,
-    )
-    chat_resp.raise_for_status()
-    response_json = chat_resp.json()
-    msg = response_json["choices"][0]["message"]
-
-    # If OpenAI requests a function call
-    if msg.get("function_call"):
-        func_name = msg["function_call"]["name"]
-        args_json = msg["function_call"]["arguments"]
-        args = json.loads(args_json)
-        # Call the correct Python function
-        if func_name in function_map:
-            function_result = function_map[func_name](**args)
-        else:
-            function_result = {"error": f"Unknown function: {func_name}"}
-        # Append function call and output to history
-        st.session_state.messages.append({
-            "role": "function",
-            "name": func_name,
-            "content": json.dumps(function_result),
-        })
-
-        # Second OpenAI call: Get final answer with function result
-        followup_messages = st.session_state.messages.copy()
-        final_resp = requests.post(
+# --- Sending user input and OpenAI call logic using a callback
+def send_message():
+    user_input = st.session_state.temp_input
+    if user_input and user_input.strip():
+        st.session_state.messages.append({"role": "user", "content": user_input})
+
+        # Compose messages for OpenAI (entire chat history)
+        chat_messages = st.session_state.messages.copy()
+
+        # First OpenAI call: Check for function call
+        chat_resp = requests.post(
             "https://api.openai.com/v1/chat/completions",
             headers=HEADERS,
             json={
                 "model": "gpt-3.5-turbo-1106",
-                "messages": followup_messages,
+                "messages": chat_messages,
+                "functions": function_schema,
+                "function_call": "auto",
                 "temperature": 0,
-                "max_tokens": 1500,
+                "max_tokens": 1000,
             },
             timeout=60,
         )
-        final_resp.raise_for_status()
-        answer = final_resp.json()["choices"][0]["message"]["content"]
-        # Add assistant's reply to chat
-        st.session_state.messages.append({"role": "assistant", "content": answer})
-    else:
-        # No function call: Just add model's reply
-        st.session_state.messages.append({"role": "assistant", "content": msg["content"]})
+        chat_resp.raise_for_status()
+        response_json = chat_resp.json()
+        msg = response_json["choices"][0]["message"]
+
+        # If OpenAI requests a function call
+        if msg.get("function_call"):
+            func_name = msg["function_call"]["name"]
+            args_json = msg["function_call"]["arguments"]
+            args = json.loads(args_json)
+            # Call the correct Python function
+            if func_name in function_map:
+                function_result = function_map[func_name](**args)
+            else:
+                function_result = {"error": f"Unknown function: {func_name}"}
+            # Append function call and output to history
+            st.session_state.messages.append({
+                "role": "function",
+                "name": func_name,
+                "content": json.dumps(function_result),
+            })
+
+            # Second OpenAI call: Get final answer with function result
+            followup_messages = st.session_state.messages.copy()
+            final_resp = requests.post(
+                "https://api.openai.com/v1/chat/completions",
+                headers=HEADERS,
+                json={
+                    "model": "gpt-3.5-turbo-1106",
+                    "messages": followup_messages,
+                    "temperature": 0,
+                    "max_tokens": 1500,
+                },
+                timeout=60,
+            )
+            final_resp.raise_for_status()
+            answer = final_resp.json()["choices"][0]["message"]["content"]
+            # Add assistant's reply to chat
+            st.session_state.messages.append({"role": "assistant", "content": answer})
+        else:
+            # No function call: Just add model's reply
+            st.session_state.messages.append({"role": "assistant", "content": msg["content"]})
 
-    # Clear input after sending (no session state error!)
-    st.session_state.temp_input = ""
-    # The UI will update automatically
+        # Clear input after sending (now legal and safe)
+        st.session_state.temp_input = ""
 
+# --- User input box at bottom (like ChatGPT)
+if df is not None:
+    st.text_input("Your message:", key="temp_input", on_change=send_message)