Update app.py

app.py

@@ -183,12 +183,8 @@ def process_input():
                                 'result': result_str
                             })
 
-                            # Append the assistant's insights to the history
-                            st.session_state.history.append({"role": "assistant", "content": insights})
-                            
-                            # Displaying insights inside an expander to avoid cutoff
-                            with st.expander("Click to see detailed insights"):
-                                st.markdown(insights)
+                            # Use st.text_area to avoid text cutoff for long outputs
+                            st.text_area("Insights", value=insights, height=300)
 
                             # Append the result DataFrame to the history
                             st.session_state.history.append({"role": "assistant", "content": result})
@@ -204,12 +200,8 @@ def process_input():
                         'result': dataset_summary
                     })
 
-                    # Append the assistant's recommendations to the history
-                    st.session_state.history.append({"role": "assistant", "content": recommendations})
-
-                    # Displaying recommendations inside an expander to avoid cutoff
-                    with st.expander("Click to see detailed recommendations"):
-                        st.markdown(recommendations)
+                    # Use st.text_area to avoid text cutoff for long outputs
+                    st.text_area("Recommendations", value=recommendations, height=300)
                     
                 else:
                     # Generate insights based on general insights (without recommendations)
@@ -218,12 +210,8 @@ def process_input():
                         'result': dataset_summary
                     })
 
-                    # Append the assistant's general insights to the history
-                    st.session_state.history.append({"role": "assistant", "content": insights})
-
-                    # Displaying insights inside an expander to avoid cutoff
-                    with st.expander("Click to see detailed insights"):
-                        st.markdown(insights)
+                    # Use st.text_area to avoid text cutoff for long outputs
+                    st.text_area("Insights", value=insights, height=300)
         
         except Exception as e:
             logging.error(f"An error occurred: {e}")
@@ -232,19 +220,3 @@ def process_input():
 
         # Reset the user_input in session state
         st.session_state['user_input'] = ''
-
-    
-
-# Display the conversation history
-for message in st.session_state.history:
-    if message['role'] == 'user':
-        st.markdown(f"**User:** {message['content']}")
-    elif message['role'] == 'assistant':
-        if isinstance(message['content'], pd.DataFrame):
-            st.markdown("**Assistant:** Query Results:")
-            st.dataframe(message['content'])
-        else:
-            st.markdown(f"**Assistant:** {message['content']}")
-
-# Place the input field at the bottom with the callback
-st.text_input("Enter your message:", key='user_input', on_change=process_input)