Update app.py

app.py

@@ -34,10 +34,19 @@ if user_input := st.chat_input("What do you want to ask?"):
         model=selected_model,
         messages=st.session_state.messages
     )
-    
+
+    # Check and handle response type
+    if isinstance(response, str):
+        response_content = response  # Directly use if it's a string
+    else:
+        try:
+            response_content = response['choices'][0]['message']['content']
+        except (IndexError, KeyError):
+            response_content = "Error: Unexpected response structure."
+
     # Display assistant response
     with st.chat_message("assistant"):
-        st.markdown(response['choices'][0]['message']['content'])
-    
+        st.markdown(response_content)
+
     # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": response['choices'][0]['message']['content']})
+    st.session_state.messages.append({"role": "assistant", "content": response_content})