Added trendline option in the scatterplot and associated update to the project requirements file for deployent on HF spaces

pages/040_Test_Reporter.py

@@ -51,9 +51,16 @@ def show_stats(for_test_group: str):
                 for elapsed, resp_len in zip(arch['elapsed'], arch['response_len']):
                     data.append([arch['arch_name'], elapsed / 1000, resp_len])
             df = pd.DataFrame(data, columns=['Architecture', 'Elapsed time', 'Response length'])
-            fig = px.scatter(df, x='Elapsed time', y='Response length', color='Architecture')
+
+            chart_area = st.container()
+            if st.checkbox("Show regression lines"):
+                fig = px.scatter(df, x='Elapsed time', y='Response length', color='Architecture', trendline='ols', trendline_color_override='red')
+
+            else:
+                fig = px.scatter(df, x='Elapsed time', y='Response length', color='Architecture')
             fig.update_xaxes(tickangle=-90)
-            st.plotly_chart(fig, use_container_width=True)
+            with chart_area:
+                st.plotly_chart(fig, use_container_width=True)
 
     def show_q_and_a_detail(stats):
         questions = list(stats[0]['q_and_a'].keys())

requirements.txt

@@ -3,3 +3,4 @@ plotly==5.17.0
 chromadb==0.4.15
 better-profanity==0.7.0
 regex==2023.10.3
+statsmodels==0.14.1