Update app.py

app.py

@@ -156,6 +156,19 @@ def plt_to_html(fig):
     plt.close(fig)
     return f'<img src="data:image/png;base64,{img_str}" alt="Chart">'
 
+def clear_analytics():
+    """Clear all analytics data by removing the log file"""
+    log_file = "analytics/chat_log.json"
+    
+    if os.path.exists(log_file):
+        try:
+            os.remove(log_file)
+            return "Analytics data cleared successfully."
+        except Exception as e:
+            return f"Error clearing analytics: {str(e)}"
+    else:
+        return "No analytics data to clear."
+
 def generate_analytics():
     """Generate analytics from the chat log"""
     log_file = "analytics/chat_log.json"
@@ -315,12 +328,12 @@ with gr.Blocks(title="Groq AI Chat Playground") as app:
         with gr.Tab("Analytics Dashboard"):
             with gr.Column():
                 gr.Markdown("# Usage Analytics Dashboard")
-                refresh_analytics_button = gr.Button("Refresh Analytics")
                 
-                gr.Markdown("""
-                **Note**: Please reload the site to clear analytics data.
-                """)
+                with gr.Row():
+                    refresh_analytics_button = gr.Button("Refresh Analytics")
+                    clear_analytics_button = gr.Button("Clear Analytics", variant="secondary")
                 
+                analytics_status = gr.Markdown()
                 analytics_summary = gr.Markdown()
                 
                 with gr.Row():
@@ -370,7 +383,17 @@ with gr.Blocks(title="Groq AI Chat Playground") as app:
         inputs=[],
         outputs=[analytics_summary, model_usage_chart, response_time_chart, analytics_table]
     )
+    
+    clear_analytics_button.click(
+        fn=clear_analytics,
+        inputs=[],
+        outputs=[analytics_status]
+    ).then(
+        fn=generate_analytics,
+        inputs=[],
+        outputs=[analytics_summary, model_usage_chart, response_time_chart, analytics_table]
+    )
 
 # Launch the app
 if __name__ == "__main__":
-    app.launch(share=False)
+    app.launch(share=False)