Merge pull request #17 from robertmcleod2/chain-assistant-final-version

put plots in consistent position
robertmcleod2 · Oct 14, 2024 · cc64f47 · cc64f47
2 parents b254d31 + 2385af0
commit cc64f47
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/chatbot_rag_anomaly_detection.py b/src/chatbot_rag_anomaly_detection.py
@@ -40,9 +40,7 @@ def __init__(self):
             analyse_weather_data(weather_df, anomalies, prolonged_anomalies)
         )
         fig = plot_anomalies(df, anomalies, prolonged_anomalies)
-        st.session_state.messages.append({"role": "assistant", "content": fig})
         fig_weather = plot_weather(weather_df)
-        st.session_state.messages.append({"role": "assistant", "content": fig_weather})
         self.anomaly_text = generate_anomaly_text(
             anomalies,
             prolonged_anomalies,
@@ -61,6 +59,8 @@ def __init__(self):
         st.plotly_chart(fig)
         st.plotly_chart(fig_weather)
         st.session_state.messages.append({"role": "assistant", "content": response})
+        st.session_state.messages.append({"role": "assistant", "content": fig})
+        st.session_state.messages.append({"role": "assistant", "content": fig_weather})
 
     def initialize_chain(self):