Update app.py

app.py

@@ -40,10 +40,9 @@ def get_streamed_response(message, history):
         "content": "From now on, you are an AI assistant knowledgeable in general topics. You can respond with relevant information and provide concise, friendly replies. Always maintain a helpful and neutral tone. Ensure to be concise to keep the conversation flowing smoothly."
     }]
     
-    for human, assistant in history:
-        all_message.append({"role": "user", "content": human})
-        all_message.append({"role": "assistant", "content": assistant})
-
+    for msg in history:
+        all_message.append(msg)
+    all_message.append({"role": "user", "content": message})
     return all_message
 
 # Display chat messages from history on app rerun
@@ -61,7 +60,7 @@ if prompt := st.chat_input("Type your message here..."):
     st.session_state.messages.append({"role": "user", "content": prompt})
 
     # Prepare all messages for the conversation context
-    history = [(msg["content"], next((m["content"] for m in st.session_state.messages if m["role"] == "assistant"), "")) for msg in st.session_state.messages]
+    history = st.session_state.messages
 
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
@@ -73,12 +72,15 @@ if prompt := st.chat_input("Type your message here..."):
                 max_tokens=150  # Adjust the token limit as needed
             )
 
-            st.markdown(response['choices'][0]['message']['content'])
-
+            # Ensure response handling is safe
+            if 'choices' in response and response['choices']:
+                st.markdown(response['choices'][0]['message']['content'])
+                st.session_state.messages.append({"role": "assistant", "content": response['choices'][0]['message']['content']})
+            else:
+                st.markdown("No response received from the assistant.")
+        
         except Exception as e:
             st.markdown("An error occurred. Please try again later.")
             st.markdown("Error details:")
             st.markdown(str(e))
-
-    # Add assistant's 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": "An error occurred. Please try again later."})