Update app.py

app.py

@@ -90,14 +90,13 @@ if prompt := st.chat_input("Type your message..."):
             # Handle API response
             if isinstance(output, list) and len(output) > 0 and 'generated_text' in output[0]:
                 assistant_response = output[0]['generated_text']
-            else:
-                st.error("Error: Unable to generate a response. Please try again.")
-                continue  # Skip further execution for this iteration
 
-            with st.chat_message("assistant"):
-                st.markdown(assistant_response)
+                with st.chat_message("assistant"):
+                    st.markdown(assistant_response)
 
-            st.session_state.messages.append({"role": "assistant", "content": assistant_response})
+                st.session_state.messages.append({"role": "assistant", "content": assistant_response})
+            else:
+                st.error("Error: Unable to generate a response. Please try again.")
 
     except Exception as e:
-        st.error(f"Application Error: {str(e)}")
+        st.error(f"Application Error: {str(e)}")