Update services/analytics_tab_module.py

services/analytics_tab_module.py

@@ -1,15 +1,16 @@
-# analytics_tab_module.py
 import gradio as gr
 import pandas as pd
 import logging
 import time
-from datetime import datetime, date, timedelta
+from datetime import datetime, timedelta
 import numpy as np
 from collections import OrderedDict, defaultdict
 import asyncio
 import matplotlib # Keep this if create_placeholder_plot or other plot fns need it
 matplotlib.use('Agg') # Keep if necessary for plotting functions
 import matplotlib.pyplot as plt # Keep if necessary
+# ADDED: Import the Calendar component
+from gradio_calendar import Calendar
 
 # It's assumed that PLOT_CONFIGS is specific to this analytics tab.
 # If it's used elsewhere, it might need to be passed in or imported from a central config.
@@ -34,6 +35,7 @@ PLOT_CONFIGS = [
     {"label": "Volume Menzioni nel Tempo (Dettaglio)", "id": "mention_analysis_volume", "section": "Analisi Menzioni (Dettaglio)"},
     {"label": "Ripartizione Menzioni per Sentiment (Dettaglio)", "id": "mention_analysis_sentiment", "section": "Analisi Menzioni (Dettaglio)"}
 ]
+
 assert len(PLOT_CONFIGS) == 19, "Mancata corrispondenza in PLOT_CONFIGS e grafici attesi."
 
 UNIQUE_ORDERED_SECTIONS = list(OrderedDict.fromkeys(pc["section"] for pc in PLOT_CONFIGS))
@@ -45,7 +47,6 @@ class AnalyticsTab:
                  plot_id_to_formula_map, plot_formulas_data, icons,
                  fn_build_plot_area, fn_update_plot_figures, fn_create_placeholder_plot,
                  fn_get_initial_insight, fn_generate_llm_response):
-
         # Shared Gradio states passed from the main app
         self.token_state = token_state
         self.chat_histories_st = chat_histories_st
@@ -99,37 +100,14 @@ class AnalyticsTab:
         is_custom = selection == "Intervallo Personalizzato"
         return gr.update(visible=is_custom), gr.update(visible=is_custom)
 
-    # --- NEW HELPER FUNCTION ---
-    # This function handles the conversion from a date object (from gr.Date)
-    # to a datetime object, which the plotting function likely expects.
-    def _handle_date_input(self, date_val):
-        """Converts various date inputs to a timezone-aware datetime object."""
-        if date_val is None:
-            return None
-        if isinstance(date_val, datetime):
-            return date_val # Already a datetime object
-        if isinstance(date_val, date):
-            # Combine the date with the beginning of the day to create a datetime object
-            return datetime.combine(date_val, datetime.min.time())
-        # Add handling for string if necessary, though gr.Date usually returns date/datetime objects
-        if isinstance(date_val, str):
-            try:
-                # Attempt to parse a date string like 'YYYY-MM-DD'
-                parsed_date = datetime.strptime(date_val, '%Y-%m-%d').date()
-                return datetime.combine(parsed_date, datetime.min.time())
-            except ValueError:
-                logging.warning(f"Could not parse date string: {date_val}")
-                return None
-        return None
-
     async def _handle_panel_action(
         self, plot_id_clicked: str, action_type: str, current_active_action_from_state: dict,
         current_chat_histories: dict, current_chat_plot_id: str,
         current_plot_data_for_chatbot: dict, current_explored_plot_id: str
     ):
         logging.info(f"Panel Action: '{action_type}' for plot '{plot_id_clicked}'. Active: {current_active_action_from_state}, Explored: {current_explored_plot_id}")
-        clicked_plot_config = next((p for p in PLOT_CONFIGS if p["id"] == plot_id_clicked), None)
 
+        clicked_plot_config = next((p for p in PLOT_CONFIGS if p["id"] == plot_id_clicked), None)
         if not clicked_plot_config:
             logging.error(f"Config not found for plot_id {plot_id_clicked}")
             num_plots = len(PLOT_CONFIGS)
@@ -144,6 +122,7 @@ class AnalyticsTab:
 
         clicked_plot_label = clicked_plot_config["label"]
         clicked_plot_section = clicked_plot_config["section"]
+
         hypothetical_new_active_state = {"plot_id": plot_id_clicked, "type": action_type}
         is_toggling_off = current_active_action_from_state == hypothetical_new_active_state
 
@@ -191,7 +170,6 @@ class AnalyticsTab:
             
             if action_type == "insights": # Specifically when closing insights chat
                 new_current_chat_plot_id = None # Clear the chat context
-
         else: # Toggling ON a panel action
             new_active_action_state_to_set = hypothetical_new_active_state
             action_col_visible_update = gr.update(visible=True)
@@ -211,16 +189,15 @@ class AnalyticsTab:
                 is_active_formula = new_active_action_state_to_set == {"plot_id": cfg_btn["id"], "type": "formula"}
                 generated_bomb_btn_updates.append(gr.update(value=self.ACTIVE_ICON if is_active_insights else self.BOMB_ICON))
                 generated_formula_btn_updates.append(gr.update(value=self.ACTIVE_ICON if is_active_formula else self.FORMULA_ICON))
-
+            
             if action_type == "insights":
                 insights_chatbot_visible_update = gr.update(visible=True)
                 insights_chat_input_visible_update = gr.update(visible=True)
                 insights_suggestions_row_visible_update = gr.update(visible=True)
                 new_current_chat_plot_id = plot_id_clicked # Set chat context to this plot
-
                 history = current_chat_histories.get(plot_id_clicked, [])
                 summary = current_plot_data_for_chatbot.get(plot_id_clicked, f"No summary available for '{clicked_plot_label}'.")
-
+                
                 if not history: # First time opening chat for this plot
                     prompt, sugg = self.get_initial_insight_prompt_and_suggestions(plot_id_clicked, clicked_plot_label, summary)
                     llm_history_for_api = [{"role": "user", "content": prompt}] # API expects list of dicts
@@ -229,7 +206,7 @@ class AnalyticsTab:
                     updated_chat_histories = {**current_chat_histories, plot_id_clicked: history}
                 else: # Re-opening chat, just get new suggestions if any
                     _, sugg = self.get_initial_insight_prompt_and_suggestions(plot_id_clicked, clicked_plot_label, summary)
-
+                
                 chatbot_content_update = gr.update(value=history)
                 s1_upd = gr.update(value=sugg[0] if sugg and len(sugg) > 0 else "N/A")
                 s2_upd = gr.update(value=sugg[1] if sugg and len(sugg) > 1 else "N/A")
@@ -251,26 +228,26 @@ class AnalyticsTab:
 
         # Order of updates must match self.action_panel_outputs_list
         final_updates = [
-            action_col_visible_update,       # global_actions_column_ui
-            insights_chatbot_visible_update,   # insights_chatbot_ui (visibility)
-            chatbot_content_update,            # insights_chatbot_ui (content)
+            action_col_visible_update,        # global_actions_column_ui
+            insights_chatbot_visible_update,    # insights_chatbot_ui (visibility)
+            chatbot_content_update,             # insights_chatbot_ui (content)
             insights_chat_input_visible_update, # insights_chat_input_ui
             insights_suggestions_row_visible_update, # insights_suggestions_row_ui
-            s1_upd, s2_upd, s3_upd,             # suggestion buttons
-            formula_display_visible_update,    # formula_display_markdown_ui (visibility)
-            formula_content_update,            # formula_display_markdown_ui (content)
-            formula_close_hint_visible_update, # formula_close_hint_md
-            new_active_action_state_to_set,    # active_panel_action_state
-            new_current_chat_plot_id,          # current_chat_plot_id_st
-            updated_chat_histories,            # chat_histories_st
-            new_explored_plot_id_to_set        # explored_plot_id_state
+            s1_upd, s2_upd, s3_upd,              # suggestion buttons
+            formula_display_visible_update,     # formula_display_markdown_ui (visibility)
+            formula_content_update,             # formula_display_markdown_ui (content)
+            formula_close_hint_visible_update,  # formula_close_hint_md
+            new_active_action_state_to_set,     # active_panel_action_state
+            new_current_chat_plot_id,           # current_chat_plot_id_st
+            updated_chat_histories,             # chat_histories_st
+            new_explored_plot_id_to_set         # explored_plot_id_state
         ]
         final_updates.extend(generated_panel_vis_updates) # Plot panel visibilities
         final_updates.extend(generated_bomb_btn_updates)    # Bomb button icons
         final_updates.extend(generated_formula_btn_updates) # Formula button icons
         final_updates.extend(generated_explore_btn_updates) # Explore button icons
         final_updates.extend(section_title_vis_updates)   # Section title visibilities
-
+        
         logging.debug(f"handle_panel_action returning {len(final_updates)} updates. Expected {15 + 4*len(PLOT_CONFIGS) + NUM_UNIQUE_SECTIONS}.")
         return final_updates
 
@@ -286,19 +263,17 @@ class AnalyticsTab:
 
         history_for_api = chat_histories.get(current_plot_id, []).copy()
         if not isinstance(history_for_api, list): history_for_api = []
-
         history_for_api.append({"role": "user", "content": user_message})
-        
+                
         current_display_history = history_for_api.copy()
-
         yield current_display_history, gr.update(value=""), chat_histories
 
         assistant_response = await self.generate_llm_response(user_message, current_plot_id, plot_label, history_for_api, summary_data)
-        history_for_api.append({"role": "assistant", "content": assistant_response})
 
+        history_for_api.append({"role": "assistant", "content": assistant_response})
         updated_chat_histories = {**chat_histories, current_plot_id: history_for_api}
         current_display_history.append({"role": "assistant", "content": assistant_response})
-
+        
         yield current_display_history, "", updated_chat_histories
 
     async def _handle_suggested_question_click(self, suggestion_text: str, current_plot_id: str, chat_histories: dict, current_plot_data_for_chatbot: dict):
@@ -306,7 +281,7 @@ class AnalyticsTab:
             current_history_for_plot = chat_histories.get(current_plot_id, [])
             yield current_history_for_plot, gr.update(value=""), chat_histories
             return
-        
+                
         async for update_chunk in self._handle_chat_message_submission(suggestion_text, current_plot_id, chat_histories, current_plot_data_for_chatbot):
             yield update_chunk
 
@@ -324,11 +299,11 @@ class AnalyticsTab:
 
         new_explored_id_to_set = None
         is_toggling_off_explore = (plot_id_clicked == current_explored_plot_id_from_state)
-
+        
         action_col_upd = gr.update()
         new_active_panel_state_upd = current_active_panel_action_state
         formula_hint_upd = gr.update(visible=False)
-
+        
         panel_vis_updates = []
         explore_btns_updates = []
         bomb_btns_updates = [gr.update()] * num_plots
@@ -349,7 +324,6 @@ class AnalyticsTab:
         else:
             new_explored_id_to_set = plot_id_clicked
             logging.info(f"Exploring {plot_id_clicked}. Hiding other plots/sections.")
-
             for i, sec_name in enumerate(UNIQUE_ORDERED_SECTIONS):
                 section_title_vis_updates[i] = gr.update(visible=(sec_name == section_of_clicked_plot))
 
@@ -357,7 +331,7 @@ class AnalyticsTab:
                 is_target_plot = (cfg["id"] == new_explored_id_to_set)
                 panel_vis_updates.append(gr.update(visible=is_target_plot))
                 explore_btns_updates.append(gr.update(value=self.ACTIVE_ICON if is_target_plot else self.EXPLORE_ICON))
-
+            
             if current_active_panel_action_state:
                 logging.info("Closing active insight/formula panel due to explore click.")
                 action_col_upd = gr.update(visible=False)
@@ -377,7 +351,7 @@ class AnalyticsTab:
         final_explore_updates.extend(bomb_btns_updates)
         final_explore_updates.extend(formula_btns_updates)
         final_explore_updates.extend(section_title_vis_updates)
-
+        
         logging.debug(f"handle_explore_click returning {len(final_explore_updates)} updates. Expected {4 + 4*len(PLOT_CONFIGS) + NUM_UNIQUE_SECTIONS}.")
         return final_explore_updates
 
@@ -385,21 +359,13 @@ class AnalyticsTab:
         async def _handler(curr_active_val, curr_chats_val, curr_chat_pid, curr_plot_data, curr_explored_id):
             return await self._handle_panel_action(p_id, action_type_str, curr_active_val, curr_chats_val, curr_chat_pid, curr_plot_data, curr_explored_id)
         return _handler
-    
-    # =========================================================================
-    # == MODIFIED FUNCTION: _refresh_analytics_graphs_ui
-    # =========================================================================
+        
     async def _refresh_analytics_graphs_ui(self, current_token_state_val, date_filter_val, custom_start_val, custom_end_val, current_chat_histories_val):
         logging.info("Refreshing analytics graph UI elements and resetting actions/chat (within module).")
         start_time = time.time()
-
-        # --- MODIFICATION: Use the helper function to convert date inputs ---
-        # This ensures that the `update_analytics_plots_figures` function receives a consistent datetime format.
-        start_date_dt = self._handle_date_input(custom_start_val)
-        end_date_dt = self._handle_date_input(custom_end_val)
-
-        plot_gen_results = self.update_analytics_plots_figures(current_token_state_val, date_filter_val, start_date_dt, end_date_dt, PLOT_CONFIGS)
         
+        plot_gen_results = self.update_analytics_plots_figures(current_token_state_val, date_filter_val, custom_start_val, custom_end_val, PLOT_CONFIGS)
+                
         try:
             expected_count = len(PLOT_CONFIGS) + 2
             if len(plot_gen_results) == expected_count:
@@ -413,18 +379,18 @@ class AnalyticsTab:
             status_msg = f"Errore: L'aggiornamento dei grafici non è riuscito a causa di un formato di dati imprevisto."
             gen_figs = [self.create_placeholder_plot("Error", f"Dati non validi {i}") for i in range(len(PLOT_CONFIGS))]
             new_summaries_for_chatbot = {}
-
+        
         all_updates = []
         # 1. Status Markdown
         all_updates.append(status_msg) # For self.analytics_status_md
-
+        
         # 2. Plot components
         if isinstance(gen_figs, list) and len(gen_figs) == len(PLOT_CONFIGS):
             all_updates.extend(gen_figs)
         else:
             logging.error(f"Mismatch in generated figures ({len(gen_figs) if isinstance(gen_figs, list) else 'N/A'}) and plot_configs ({len(PLOT_CONFIGS)})")
             all_updates.extend([self.create_placeholder_plot("Error", f"Figura mancante {i}") for i in range(len(PLOT_CONFIGS))])
-
+            
         # 3. UI Resets for Action Panel (9 components)
         all_updates.extend([
             gr.update(visible=False),  # global_actions_column_ui
@@ -435,7 +401,7 @@ class AnalyticsTab:
             gr.update(value="I dettagli sulla formula/metodologia appariranno qui.", visible=False),  # formula_display_markdown_ui
             gr.update(visible=False)  # formula_close_hint_md
         ])
-
+        
         # 4. State Resets (4 states)
         all_updates.extend([
             None,  # active_panel_action_state (reset)
@@ -443,28 +409,30 @@ class AnalyticsTab:
             {},    # chat_histories_st (reset all chats on refresh)
             new_summaries_for_chatbot  # plot_data_for_chatbot_st (update with new summaries)
         ])
-        
+                
         # 5. Plot-specific UI Resets (4 components per plot)
         for _ in PLOT_CONFIGS:
             all_updates.extend([
                 gr.update(value=self.BOMB_ICON),    # bomb_button
                 gr.update(value=self.FORMULA_ICON), # formula_button
                 gr.update(value=self.EXPLORE_ICON), # explore_button
-                gr.update(visible=True)            # panel_component (plot visibility itself)
+                gr.update(visible=True)           # panel_component (plot visibility itself)
             ])
-
+            
         # 6. Explored Plot ID State Reset (1 state)
         all_updates.append(None)  # explored_plot_id_state (reset)
-
+        
         # 7. Section Title Visibilities
         all_updates.extend([gr.update(visible=True)] * NUM_UNIQUE_SECTIONS)
-
+        
         end_time = time.time()
         logging.info(f"Analytics graph refresh (module) took {end_time - start_time:.2f} seconds.")
+        
         expected_len = 1 + len(PLOT_CONFIGS) + 9 + 4 + (4 * len(PLOT_CONFIGS)) + 1 + NUM_UNIQUE_SECTIONS
         logging.info(f"Prepared {len(all_updates)} updates for graph refresh. Expected {expected_len}.")
         if len(all_updates) != expected_len:
             logging.error(f"Output length mismatch in _refresh_analytics_graphs_ui: got {len(all_updates)}, expected {expected_len}")
+            
         return tuple(all_updates)
 
     def _define_callback_outputs(self):
@@ -528,8 +496,8 @@ class AnalyticsTab:
         graph_refresh_inputs = [
             self.token_state, self.date_filter_selector,
             self.custom_start_date_picker, self.custom_end_date_picker,
-            self.chat_histories_st 
-        ]
+            self.chat_histories_st
+         ]
         self.apply_filter_btn.click(
             fn=self._refresh_analytics_graphs_ui,
             inputs=graph_refresh_inputs,
@@ -537,7 +505,7 @@ class AnalyticsTab:
             show_progress="full",
             api_name="refresh_analytics_graphs_module"
         )
-
+        
         # Panel action buttons (bomb, formula, explore)
         action_click_inputs = [
             self.active_panel_action_state, self.chat_histories_st,
@@ -573,7 +541,7 @@ class AnalyticsTab:
                     )
             else:
                 logging.warning(f"UI object for plot_id '{plot_id}' not found for click handlers in module.")
-
+        
         # Chat submission
         chat_submission_outputs = [self.insights_chatbot_ui, self.insights_chat_input_ui, self.chat_histories_st]
         chat_submission_inputs = [
@@ -612,24 +580,21 @@ class AnalyticsTab:
 
     def create_tab_ui(self):
         with gr.TabItem("📊 Grafici", id="tab_analytics_module"):
-
             gr.Markdown("## 📈 Analisi Performance LinkedIn")
             gr.Markdown("Seleziona un intervallo di date per i grafici. Clicca i pulsanti (💣 Insights, ƒ Formula, 🧭 Esplora) su un grafico per azioni.")
             self.analytics_status_md = gr.Markdown("Stato analisi grafici...")
-
             with gr.Row():
                 self.date_filter_selector = gr.Radio(
                     ["Sempre", "Ultimi 7 Giorni", "Ultimi 30 Giorni", "Intervallo Personalizzato"],
                     label="Seleziona Intervallo Date per Grafici", value="Sempre", scale=3
                 )
                 with gr.Column(scale=2):
-                    # Using gr.Date to ensure the calendar popup appears reliably.
-                    # The conversion to a datetime object is handled in the _refresh_analytics_graphs_ui method.
-                    self.custom_start_date_picker = gr.Date(label="Data Inizio", visible=False)
-                    self.custom_end_date_picker = gr.Date(label="Data Fine", visible=False)
+                    # MODIFIED: Replaced gr.DateTime with the more reliable Calendar component
+                    self.custom_start_date_picker = Calendar(label="Data Inizio", visible=False, type="datetime")
+                    self.custom_end_date_picker = Calendar(label="Data Fine", visible=False, type="datetime")
 
             self.apply_filter_btn = gr.Button("🔍 Applica Filtro & Aggiorna Grafici", variant="primary")
-
+            
             self.date_filter_selector.change(
                 fn=self._toggle_custom_date_pickers,
                 inputs=[self.date_filter_selector],
@@ -668,7 +633,7 @@ class AnalyticsTab:
                         self.insights_suggestion_1_btn = gr.Button(value="Suggerimento 1", size="sm", min_width=50)
                         self.insights_suggestion_2_btn = gr.Button(value="Suggerimento 2", size="sm", min_width=50)
                         self.insights_suggestion_3_btn = gr.Button(value="Suggerimento 3", size="sm", min_width=50)
-
+                    
                     self.formula_display_markdown_ui = gr.Markdown(
                         "I dettagli sulla formula/metodologia appariranno qui.", visible=False
                     )
@@ -676,6 +641,6 @@ class AnalyticsTab:
                         "<p style='font-size:0.9em; text-align:center; margin-top:10px;'><em>Click the active ƒ button on the plot again to close this panel.</em></p>",
                         visible=False
                     )
-            
+                        
             self._define_callback_outputs()
             self._setup_callbacks()