Update app.py

app.py

@@ -163,13 +163,73 @@ def format_summary_results(summary):
         else:
             sender_stats += "<p style='color: #1f2937;'>No sender statistics available.</p>"
         
-        # Format primary abuser
-        primary_abuser = "<h3 style='color: #1f2937;'>Primary Concern</h3>"
+        # Format primary abuser section
+        primary_abuser_html = "<h3 style='color: #1f2937;'>Primary Concern</h3>"
         if summary['primary_abuser']:
-            primary_abuser += f"<p style='color: #1f2937;'>The analysis indicates that <strong style='color: #1f2937;'>{summary['primary_abuser']}</strong> "
-            primary_abuser += "shows the highest percentage of concerning communication patterns.</p>"
+            primary_abuser_html += f"<p style='color: #1f2937;'>The analysis indicates that <strong style='color: #1f2937;'>{summary['primary_abuser']}</strong> "
+            primary_abuser_html += "shows the highest percentage of concerning communication patterns.</p>"
+            
+            # Add detailed primary abuser analysis if available
+            if summary.get('primary_abuser_analysis'):
+                analysis = summary['primary_abuser_analysis']
+                
+                # Add pattern section with collapsible details
+                primary_abuser_html += "<div style='margin-top: 20px; border: 1px solid #e5e7eb; border-radius: 8px; padding: 15px; background-color: #f9fafb;'>"
+                primary_abuser_html += "<h4 style='color: #1f2937; margin-top: 0;'>Most Common Patterns</h4>"
+                
+                # Get patterns from sender stats
+                if summary['sender_stats'].get(summary['primary_abuser']):
+                    abuser_stats = summary['sender_stats'][summary['primary_abuser']]
+                    if abuser_stats.get('common_patterns'):
+                        primary_abuser_html += "<ul style='color: #1f2937;'>"
+                        for pattern, count in abuser_stats['common_patterns'][:5]:  # Top 5 patterns
+                            primary_abuser_html += f"<li style='color: #1f2937; margin-bottom: 8px;'><strong>{pattern}</strong> ({count} occurrences)</li>"
+                        primary_abuser_html += "</ul>"
+                    else:
+                        primary_abuser_html += "<p style='color: #1f2937;'>No specific patterns detected.</p>"
+                
+                # Add emotional tones section
+                primary_abuser_html += "<h4 style='color: #1f2937; margin-top: 20px;'>Emotional Tone Analysis</h4>"
+                if analysis.get('emotional_tones'):
+                    primary_abuser_html += "<ul style='color: #1f2937;'>"
+                    for tone, count in analysis['emotional_tones']:
+                        if tone != "neutral":
+                            primary_abuser_html += f"<li style='color: #1f2937;'><strong>{tone}</strong> ({count} messages)</li>"
+                    primary_abuser_html += "</ul>"
+                else:
+                    primary_abuser_html += "<p style='color: #1f2937;'>No significant emotional tones detected.</p>"
+                
+                # Add DARVO score
+                darvo_level = "High" if analysis.get('darvo_score', 0) >= 0.65 else "Moderate" if analysis.get('darvo_score', 0) >= 0.25 else "Low"
+                darvo_color = "#ef4444" if darvo_level == "High" else "#f59e0b" if darvo_level == "Moderate" else "#10b981"
+                
+                primary_abuser_html += f"""
+                <h4 style='color: #1f2937; margin-top: 20px;'>DARVO Analysis</h4>
+                <p style='color: #1f2937;'>DARVO Score: <strong style='color: {darvo_color};'>{analysis.get('darvo_score', 0):.3f}</strong> ({darvo_level})</p>
+                <p style='color: #1f2937; font-size: 0.9em;'>DARVO (Deny, Attack, Reverse Victim & Offender) indicates attempts to shift blame and manipulate the narrative.</p>
+                """
+                
+                # Add time patterns section
+                primary_abuser_html += "<h4 style='color: #1f2937; margin-top: 20px;'>Communication Timing Patterns</h4>"
+                
+                # Peak hours
+                primary_abuser_html += "<p style='color: #1f2937;'><strong>Peak Hours:</strong> "
+                if analysis.get('peak_hours'):
+                    peak_hours_formatted = [f"{h}:00" for h in analysis['peak_hours']]
+                    primary_abuser_html += f"{', '.join(peak_hours_formatted)}</p>"
+                else:
+                    primary_abuser_html += "No clear pattern</p>"
+                    
+                # Peak days
+                primary_abuser_html += "<p style='color: #1f2937;'><strong>Peak Days:</strong> "
+                if analysis.get('peak_days'):
+                    primary_abuser_html += f"{', '.join(analysis['peak_days'])}</p>"
+                else:
+                    primary_abuser_html += "No clear pattern</p>"
+                
+                primary_abuser_html += "</div>"
         else:
-            primary_abuser += "<p style='color: #1f2937;'>No clear pattern of concerning communication from a specific sender.</p>"
+            primary_abuser_html += "<p style='color: #1f2937;'>No clear pattern of concerning communication from a specific sender.</p>"
         
         # Format escalation data
         escalation = "<h3 style='color: #1f2937;'>Escalation Analysis</h3>"
@@ -180,16 +240,25 @@ def format_summary_results(summary):
             future_prediction = summary['escalation_data'].get('future_prediction')
             cyclic_pattern = summary['escalation_data'].get('cyclic_pattern', False)
             
-            escalation += f"<p style='color: #1f2937;'><strong style='color: #1f2937;'>Trend Direction:</strong> {trend.title()}</p>"
+            # Add color coding for trend direction
+            trend_color = "#ef4444" if trend == "escalating" else "#10b981" if trend == "de-escalating" else "#6b7280"
+            
+            escalation += f"<p style='color: #1f2937;'><strong style='color: #1f2937;'>Trend Direction:</strong> <span style='color: {trend_color};'>{trend.title()}</span></p>"
             escalation += f"<p style='color: #1f2937;'><strong style='color: #1f2937;'>Trend Strength:</strong> {trend_strength:.2f}</p>"
             escalation += f"<p style='color: #1f2937;'><strong style='color: #1f2937;'>Recent Average:</strong> {recent_avg:.1f}%</p>"
             
             if future_prediction is not None:
-                escalation += f"<p style='color: #1f2937;'><strong style='color: #1f2937;'>7-Day Prediction:</strong> {future_prediction:.1f}%</p>"
+                future_color = "#ef4444" if future_prediction > 70 else "#f59e0b" if future_prediction > 50 else "#10b981"
+                escalation += f"<p style='color: #1f2937;'><strong style='color: #1f2937;'>7-Day Prediction:</strong> <span style='color: {future_color};'>{future_prediction:.1f}%</span></p>"
             
             if cyclic_pattern:
                 cycle_period = summary['escalation_data'].get('cycle_period', 'unknown')
-                escalation += f"<p style='color: #1f2937;'><strong style='color: #1f2937;'>Cyclic Pattern Detected:</strong> Yes (approximately {cycle_period} days)</p>"
+                escalation += f"""
+                <div style='margin-top: 15px; padding: 10px; background-color: #fffbeb; border-left: 4px solid #f59e0b; border-radius: 4px;'>
+                    <p style='color: #1f2937; margin: 0;'><strong style='color: #1f2937;'>⚠️ Cyclic Abuse Pattern Detected</strong></p>
+                    <p style='color: #1f2937; margin-top: 5px;'>The analysis shows a repeating cycle of approximately {cycle_period} days. This is consistent with the cycle of abuse, where periods of tension and abuse are followed by reconciliation and calm before repeating.</p>
+                </div>
+                """
         else:
             escalation += "<p style='color: #1f2937;'>No escalation data available.</p>"
         
@@ -198,14 +267,14 @@ def format_summary_results(summary):
         if summary['recommendations']:
             recommendations += "<div style='max-height: 300px; overflow-y: auto;'>"
             for rec in summary['recommendations']:
-                recommendations += f"<div style='margin-bottom: 15px; padding: 10px; background: #f8f9fa; border-radius: 5px;'>"
-                recommendations += f"<h4 style='color: #1f2937;'>{rec['title']}</h4>"
+                recommendations += f"<div style='margin-bottom: 15px; padding: 15px; background: #f8f9fa; border-radius: 8px; border-left: 4px solid #3b82f6;'>"
+                recommendations += f"<h4 style='color: #1f2937; margin-top: 0;'>{rec['title']}</h4>"
                 recommendations += f"<p style='color: #1f2937;'>{rec['description']}</p>"
                 
                 if rec['actions']:
-                    recommendations += "<ul>"
+                    recommendations += "<ul style='color: #1f2937;'>"
                     for action in rec['actions']:
-                        recommendations += f"<li style='color: #1f2937;'>{action}</li>"
+                        recommendations += f"<li style='color: #1f2937; margin-bottom: 5px;'>{action}</li>"
                     recommendations += "</ul>"
                 
                 recommendations += "</div>"
@@ -218,7 +287,7 @@ def format_summary_results(summary):
         <div style="padding: 20px; border: 1px solid #ddd; border-radius: 10px; background-color: #ffffff;">
             {basic_info}
             {sender_stats}
-            {primary_abuser}
+            {primary_abuser_html}
             {escalation}
             {recommendations}
         </div>
@@ -227,9 +296,11 @@ def format_summary_results(summary):
         return html
     except Exception as e:
         logger.error(f"Error formatting summary results: {e}")
+        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:
@@ -276,8 +347,8 @@ def create_interface():
                 with gr.Column(scale=2):
                     summary_output = gr.HTML(
                         label="Analysis Summary",
-                        value="<p>Upload a file and click 'Analyze Chat History' to see results.</p>"
-                    )
+                        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
             
             with gr.Row():
                 with gr.Column():
@@ -341,54 +412,62 @@ def create_interface():
             </div>
             """)
         
-        # Define event handlers
         def handle_file_analysis(file_path):
-            if not file_path:
-                return (
-                    "<p>Please upload a file first.</p>",
-                    None, None, None, None,
-                    "<p>Please upload a file first.</p>"
-                )
-            
-            try:
-                # Analyze file
-                results_df, summary = analyze_file(file_path)
-                
-                if "error" in summary:
-                    return (
-                        f"<p>Error: {summary['error']}</p>",
-                        None, None, None, None,
-                        "<p>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'])
-                
-                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>Error analyzing file: {str(e)}</p>",
-                    None, None, None, None,
-                    "<p>Error analyzing file.</p>"
-                )
+    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():