Upload trend_utils.py

src/trend_utils.py

@@ -201,6 +201,7 @@ def get_plot(df: pd.DataFrame, start_date: str = '2023-06-01', end_date: str = '
     open_dip = plot_kwargs['open_dip']
     comm_dip = plot_kwargs['comm_dip']
     height = plot_kwargs['height']
+    custom_width = plot_kwargs['custom_width']
 
     mobile_view = True if plot_kwargs['mobile_view'] else False
 
@@ -384,10 +385,15 @@ def get_plot(df: pd.DataFrame, start_date: str = '2023-06-01', end_date: str = '
                             bordercolor='rgba(0,0,0,0.05)')
             )
 
+    if custom_width:
+        # Use a custom width for loading the graph for the first time
+        # Gets squished on initial loading as container width defaults to None while rendering for the first time
+        fig.update_layout(width=custom_width)
+
     return fig
 
 
-def get_final_trend_plot(benchmark: str = "Text", mobile_view: bool = False) -> go.Figure:
+def get_final_trend_plot(benchmark: str = "Text", mobile_view: bool = False, custom_width=None) -> go.Figure:
     """Fetch and generate the final trend plot for all models.
 
     Args:
@@ -416,7 +422,8 @@ def get_final_trend_plot(benchmark: str = "Text", mobile_view: bool = False) ->
     else:
         height = 1000
 
-    plot_kwargs = {'height': height, 'open_dip': 0, 'comm_dip': 0,
+
+    plot_kwargs = {'height': height, 'custom_width': custom_width, 'open_dip': 0, 'comm_dip': 0,
                    'mobile_view': mobile_view}
 
     benchmark_ticks = {}