Update app.py

app.py

@@ -71,6 +71,17 @@ def analyze_single_message(message):
             "risk_level": "Unknown"
         }
 
+def get_risk_color(risk_level):
+    """Get color for risk level"""
+    colors = {
+        "Critical": "#ef4444",
+        "High": "#f97316",
+        "Moderate": "#f59e0b",
+        "Low": "#10b981",
+        "Unknown": "#6b7280"
+    }
+    return colors.get(risk_level, "#6b7280")
+
 def format_analysis_results(analysis):
     """Format analysis results for display"""
     try:
@@ -129,7 +140,6 @@ def format_analysis_results(analysis):
         logger.error(f"Error formatting analysis results: {e}")
         return f"<p style='color: #1f2937;'>Error formatting results: {str(e)}</p>"
 
-
 def format_summary_results(summary):
     """Format summary results for display"""
     try:
@@ -299,8 +309,6 @@ def format_summary_results(summary):
         logger.error(traceback.format_exc())
         return f"<p style='color: #1f2937;'>Error formatting summary: {str(e)}</p>"
 
-
-
 def format_safety_plan(safety_plan):
     """Format safety plan for display"""
     try:
@@ -324,6 +332,81 @@ def format_safety_plan(safety_plan):
         logger.error(f"Error formatting safety plan: {e}")
         return f"<p style='color: #1f2937;'>Error formatting safety plan: {str(e)}</p>"
 
+def handle_file_analysis(file_path):
+    if not file_path:
+        return (
+            "<p style='color: #1f2937;'>Please upload a file first.</p>",
+            None, None, None, None,
+            "<p style='color: #1f2937;'>Please upload a file first.</p>"
+        )
+    
+    try:
+        # Analyze file
+        results_df, summary = analyze_file(file_path)
+        
+        if "error" in summary:
+            return (
+                f"<p style='color: #1f2937;'>Error: {summary['error']}</p>",
+                None, None, None, None,
+                "<p style='color: #1f2937;'>Error analyzing file.</p>"
+            )
+        
+        # Format summary
+        summary_html = format_summary_results(summary)
+        
+        # Generate plots
+        timeline = generate_timeline_chart(results_df)
+        pattern_freq = generate_pattern_frequency_chart(results_df)
+        sender_comp = generate_sender_comparison_chart(results_df)
+        time_heatmap = generate_time_of_day_heatmap(results_df)
+        
+        # Format safety plan
+        safety_plan_html = format_safety_plan(summary['safety_plan'])
+        
+        # Debug output
+        logger.info(f"Summary HTML length: {len(summary_html)}")
+        logger.info(f"Generated {len(results_df)} results")
+        logger.info(f"Timeline chart generated: {timeline is not None}")
+        logger.info(f"Pattern frequency chart generated: {pattern_freq is not None}")
+        logger.info(f"Sender comparison chart generated: {sender_comp is not None}")
+        logger.info(f"Time heatmap generated: {time_heatmap is not None}")
+        
+        return (
+            summary_html,
+            timeline,
+            pattern_freq,
+            sender_comp,
+            time_heatmap,
+            safety_plan_html
+        )
+    except Exception as e:
+        logger.error(f"Error in handle_file_analysis: {e}")
+        logger.error(traceback.format_exc())
+        return (
+            f"<p style='color: #1f2937;'>Error analyzing file: {str(e)}</p>",
+            None, None, None, None,
+            "<p style='color: #1f2937;'>Error analyzing file.</p>"
+        )
+
+def handle_message_analysis(message):
+    if not message.strip():
+        return "<p style='color: #1f2937;'>Please enter a message first.</p>"
+    
+    try:
+        # Analyze message
+        analysis = analyze_single_message(message)
+        
+        if "error" in analysis:
+            return f"<p style='color: #1f2937;'>Error: {analysis['error']}</p>"
+        
+        # Format analysis
+        html = format_analysis_results(analysis)
+        
+        return html
+    except Exception as e:
+        logger.error(f"Error in handle_message_analysis: {e}")
+        logger.error(traceback.format_exc())
+        return f"<p style='color: #1f2937;'>Error analyzing message: {str(e)}</p>"
 
 def create_interface():
     """Create Gradio interface"""
@@ -348,7 +431,8 @@ def create_interface():
                     summary_output = gr.HTML(
                         label="Analysis Summary",
                         value="<p style='color: #1f2937;'>Upload a file and click 'Analyze Chat History' to see results.</p>",
-                        elem_id="summary_output")  # Add an element ID for easier debugging
+                        elem_id="summary_output"
+                    )
             
             with gr.Row():
                 with gr.Column():
@@ -367,7 +451,7 @@ def create_interface():
             with gr.Row():
                 safety_plan_output = gr.HTML(
                     label="Safety Plan",
-                    value="<p>Safety plan will appear here after analysis.</p>"
+                    value="<p style='color: #1f2937;'>Safety plan will appear here after analysis.</p>"
                 )
         
         with gr.Tab("Single Message Analysis"):
@@ -383,7 +467,7 @@ def create_interface():
                 with gr.Column(scale=2):
                     message_analysis_output = gr.HTML(
                         label="Message Analysis",
-                        value="<p>Enter a message and click 'Analyze Message' to see results.</p>"
+                        value="<p style='color: #1f2937;'>Enter a message and click 'Analyze Message' to see results.</p>"
                     )
         
         with gr.Tab("About"):
@@ -412,83 +496,6 @@ def create_interface():
             </div>
             """)
         
-        def handle_file_analysis(file_path):
-    if not file_path:
-        return (
-            "<p style='color: #1f2937;'>Please upload a file first.</p>",
-            None, None, None, None,
-            "<p style='color: #1f2937;'>Please upload a file first.</p>"
-        )
-    
-    try:
-        # Analyze file
-        results_df, summary = analyze_file(file_path)
-        
-        if "error" in summary:
-            return (
-                f"<p style='color: #1f2937;'>Error: {summary['error']}</p>",
-                None, None, None, None,
-                "<p style='color: #1f2937;'>Error analyzing file.</p>"
-            )
-        
-        # Format summary
-        summary_html = format_summary_results(summary)
-        
-        # Generate plots
-        timeline = generate_timeline_chart(results_df)
-        pattern_freq = generate_pattern_frequency_chart(results_df)
-        sender_comp = generate_sender_comparison_chart(results_df)
-        time_heatmap = generate_time_of_day_heatmap(results_df)
-        
-        # Format safety plan
-        safety_plan_html = format_safety_plan(summary['safety_plan'])
-        
-        # Debug output
-        logger.info(f"Summary HTML length: {len(summary_html)}")
-        logger.info(f"Generated {len(results_df)} results")
-        logger.info(f"Timeline chart generated: {timeline is not None}")
-        logger.info(f"Pattern frequency chart generated: {pattern_freq is not None}")
-        logger.info(f"Sender comparison chart generated: {sender_comp is not None}")
-        logger.info(f"Time heatmap generated: {time_heatmap is not None}")
-        
-        return (
-            summary_html,
-            timeline,
-            pattern_freq,
-            sender_comp,
-            time_heatmap,
-            safety_plan_html
-        )
-    except Exception as e:
-        logger.error(f"Error in handle_file_analysis: {e}")
-        logger.error(traceback.format_exc())
-        return (
-            f"<p style='color: #1f2937;'>Error analyzing file: {str(e)}</p>",
-            None, None, None, None,
-            "<p style='color: #1f2937;'>Error analyzing file.</p>"
-        )
-
-        
-        def handle_message_analysis(message):
-            if not message.strip():
-                return "<p>Please enter a message first.</p>"
-            
-            try:
-                # Analyze message
-                analysis = analyze_single_message(message)
-                
-                if "error" in analysis:
-                    return f"<p>Error: {analysis['error']}</p>"
-                
-                # Format analysis
-                html = format_analysis_results(analysis)
-                
-                return html
-            except Exception as e:
-                logger.error(f"Error in handle_message_analysis: {e}")
-                logger.error(traceback.format_exc())
-                return f"<p>Error analyzing message: {str(e)}</p>"
-        
         # Connect event handlers
         analyze_button.click(
             handle_file_analysis,
@@ -514,3 +521,4 @@ def create_interface():
 if __name__ == "__main__":
     demo = create_interface()
     demo.launch(server_name="0.0.0.0", server_port=7860)
+