Update app.py

app.py

@@ -104,46 +104,50 @@ def run_conversation(user_prompt):
             },
         }
     ]
-    response = client.chat.completions.create(
-        model=model_option,
-        messages=messages,
-        tools=tools,
-        tool_choice="auto",
-        max_tokens=4096
-    )
+    try:
+        response = client.chat.completions.create(
+            model=model_option,
+            messages=messages,
+            tools=tools,
+            tool_choice="auto",
+            max_tokens=4096
+        )
 
-    response_message = response.choices[0].delta
-    tool_calls = response_message.tool_calls
+        response_message = response.choices[0].delta
+        tool_calls = response_message.tool_calls
 
-    if tool_calls:
-        available_functions = {
-            "time_date": get_tool_owner_info
-        }
+        if tool_calls:
+            available_functions = {
+                "time_date": get_tool_owner_info
+            }
 
-        messages.append(response_message)
-
-        for tool_call in tool_calls:
-            function_name = tool_call.function.name
-            function_to_call = available_functions[function_name]
-            function_args = json.loads(tool_call.function.arguments)
-            function_response = function_to_call(**function_args)
-            messages.append(
-                {
-                    "tool_call_id": tool_call.id,
-                    "role": "tool",
-                    "name": function_name,
-                    "content": function_response,
-                }
-            )
+            messages.append(response_message)
 
-        second_response = client.chat.completions.create(
-            model=model_option,
-            messages=messages
-        )
+            for tool_call in tool_calls:
+                function_name = tool_call.function.name
+                function_to_call = available_functions[function_name]
+                function_args = json.loads(tool_call.function.arguments)
+                function_response = function_to_call(**function_args)
+                messages.append(
+                    {
+                        "tool_call_id": tool_call.id,
+                        "role": "tool",
+                        "name": function_name,
+                        "content": function_response,
+                    }
+                )
+
+            second_response = client.chat.completions.create(
+                model=model_option,
+                messages=messages
+            )
 
-        return second_response.choices[0].delta.content
-    else:
-        return response_message.content
+            return second_response.choices[0].delta.content
+        else:
+            return response_message.content
+    except Exception as e:
+        st.error(e, icon="🚨")
+        return None
 
 def get_tool_owner_info():
     owner_info = {
@@ -157,6 +161,9 @@ if prompt := st.chat_input("Enter your prompt here..."):
     with st.chat_message("user", avatar="🕺"):  
         st.markdown(prompt)
 
+    chat_responses_generator = None
+    full_response = None
+
     try:
         chat_completion = client.chat.completions.create(
             model=model_option,
@@ -168,9 +175,8 @@ if prompt := st.chat_input("Enter your prompt here..."):
             stream=True,
         )
 
-        with st.chat_message("assistant", avatar="🤖"):
-            chat_responses_generator = generate_chat_responses(chat_completion)
-            full_response = st.write_stream(chat_responses_generator)
+        chat_responses_generator = generate_chat_responses(chat_completion)
+        full_response = st.write_stream(chat_responses_generator)
     except Exception as e:
         st.error(e, icon="🚨")
 
@@ -178,8 +184,8 @@ if prompt := st.chat_input("Enter your prompt here..."):
         st.session_state.messages.append(
             {"role": "assistant", "content": full_response}
         )
-    else:
-        combined_response = "\n".join(str(item) for item in full_response)
+    elif chat_responses_generator:
+        combined_response = "\n".join(str(item) for item in chat_responses_generator)
         st.session_state.messages.append(
             {"role": "assistant", "content": combined_response}
         )