WhatsApp Relationship Analyzer

Mix.install([
  {:nx, "~> 0.9"},
  {:axon, "~> 0.7"},
  {:explorer, "~> 0.10"},
  {:exla, "~> 0.9"},
  {:kino, "~> 0.15"},
  {:kino_vega_lite, "~> 0.1"},
  {:vega_lite, "~> 0.1"},
  {:stemmer, "~> 1.2"},
  {:whatsapp_analyser, github: "zoedsoupe/whatsapp-relationship-analyser", branch: "main"}
])

Setup

> Important: To run this notebook properly: > > 1. Click the Settings (⚙️) button in the top right > 2. Under “Runtime,” select “Mix standalone” > 3. For the “Path to mix project,” enter the path to the root of this repository > 4. Click “Apply changes”

Analysis

# Set up aliases for visualization
alias VegaLite, as: Vl
alias Kino.VegaLite, as: KVl

# Create a form with file input
form = Kino.Control.form(
  [
    file: Kino.Input.file("Upload WhatsApp Chat Export (.txt)")
  ],
  submit: "Analyze Chat"
)

# Render the form
Kino.render(form)

# Create a frame to display results
result_frame = Kino.Frame.new()
Kino.render(result_frame)

# Handle form submission
Kino.listen(form, fn %{data: %{file: file}} ->
  if file do
    # Get file content using the appropriate Kino function
    path = Kino.Input.file_path(file.file_ref)
    
    
    # Run the analysis
    Kino.Frame.render(result_frame, Kino.Markdown.new("Analyzing chat file... This may take a moment."))
    
    try do
      # Run the analysis
      analysis = WhatsAppAnalyzer.analyze_chat(path)
      
      # Display summary
      summary = WhatsAppAnalyzer.summarize_relationship(analysis)
      
      summary_text = """
      # Relationship Analysis Summary
      
      **Classification:** #{summary.classification}
      **Confidence Score:** #{summary.confidence_score}%
      **Total Messages:** #{summary.message_count}
      **Time Span:** #{round(summary.time_span_days)} days
      
      ## Key Indicators
      #{Enum.map(summary.primary_indicators, fn {indicator, score} -> "- #{indicator}: #{score}%" end) |> Enum.join("\n")}
      """
      
      Kino.Frame.render(result_frame, Kino.Markdown.new(summary_text))
      
      # Display visualizations
      Kino.Frame.append(result_frame, Kino.Markdown.new("## Message Frequency Over Time"))
      Kino.Frame.append(result_frame, KVl.new(analysis.visualizations.message_frequency))
      
      Kino.Frame.append(result_frame, Kino.Markdown.new("## Messages by Sender"))
      Kino.Frame.append(result_frame, KVl.new(analysis.visualizations.sender_distribution))
      
      Kino.Frame.append(result_frame, Kino.Markdown.new("## Activity Heatmap by Time of Day"))
      Kino.Frame.append(result_frame, KVl.new(analysis.visualizations.time_heatmap))
      
      Kino.Frame.append(result_frame, Kino.Markdown.new("## Relationship Indicators"))
      Kino.Frame.append(result_frame, KVl.new(analysis.visualizations.relationship_radar))
      
      Kino.Frame.append(result_frame, Kino.Markdown.new("## Classification Confidence"))
      Kino.Frame.append(result_frame, KVl.new(analysis.visualizations.classification))
      
      # Display detailed statistics
      Kino.Frame.append(result_frame, Kino.Markdown.new("## Detailed Statistics"))
      
      # Show top romantic indicators
      romantic_indicators = analysis.analysis.romantic_indicators
      
      romantic_text = """
      ### Romantic Language
      - Total romantic indicators: #{romantic_indicators.total_indicators}
      - Percentage of messages with romantic language: #{romantic_indicators.percentage_of_messages}%
      """
      
      Kino.Frame.append(result_frame, Kino.Markdown.new(romantic_text))
      
      # Show conversation initiation statistics
      initiations = analysis.analysis.conversation_initiation
      
      initiation_text = """
      ### Conversation Patterns
      - Total conversations: #{initiations.total_conversations}
      - Conversation initiations by sender:
      #{Enum.map(initiations.initiation_percentage, fn {sender, percentage} -> "  - #{sender}: #{percentage}%" end) |> Enum.join("\n")}
      """
      
      Kino.Frame.append(result_frame, Kino.Markdown.new(initiation_text))
      
      # Show time-based patterns
      time_patterns = analysis.analysis.time_of_day_patterns
      
      time_text = """
      ### Time Patterns
      - Most active time period: #{Enum.max_by(time_patterns.percentage_by_period, fn {_, v} -> v end) |> elem(0)}
      - Weekend vs Weekday: #{round(analysis.analysis.day_of_week_patterns.weekday_vs_weekend.weekend_percentage)}% weekend, #{round(analysis.analysis.day_of_week_patterns.weekday_vs_weekend.weekday_percentage)}% weekday
      """
      
      Kino.Frame.append(result_frame, Kino.Markdown.new(time_text))
      
    rescue
      e ->
        error_message = """
        Error analyzing chat: #{inspect(e)}
        
        Stack trace:
        #{Exception.format_stacktrace(__STACKTRACE__)}
        """
        
        Kino.Frame.render(result_frame, Kino.Markdown.new(error_message))
    end
  else
    Kino.Frame.render(result_frame, Kino.Markdown.new("Please upload a WhatsApp chat export file first."))
  end
end)