Added Harry Potter analysis

app.py

@@ -1,14 +1,31 @@
-import streamlit as st
 import pickle
+import math
+import streamlit as st
+import plotly.graph_objects as go
+import matplotlib.pyplot as plt
+import pandas as pd
+import numpy as np
 
-st.title("Contact List Analysis")
+reports = {"Hillary Clinton": {"emotions_report_filename" : "emotions_report.pkl",
+                               "goals_report_filename" : "goals_report.pkl"},
+            "Harry Potter": {"emotions_report_filename" : "hp_emotions.pkl",
+                               "goals_report_filename" : "hp_goals.pkl"}
+          }
 
-with open('emotions_report.pkl', 'rb') as file:
-    emotions_report = pickle.load(file)
+if 'character' not in st.session_state:
+    st.session_state.character = "Harry Potter"
 
-with open('goals_report.pkl', 'rb') as file:
-    goals_report = pickle.load(file)
+for ch in reports.keys():
+    with open(reports[ch]["goals_report_filename"], 'rb') as file:
+        reports[ch]["goals_report"] = pickle.load(file)
+    
+    with open(reports[ch]["emotions_report_filename"], 'rb') as file:
+        reports[ch]["emotions_report"] = pickle.load(file)
 
+emotions_report = reports[st.session_state.character]["emotions_report"]
+goals_report = reports[st.session_state.character]["goals_report"]
+
+st.title(f"{st.session_state.character}'s Contact List Analysis")
 emotions_tab, goals_tab = st.tabs(["Emotions", "Goals"])
 
 with emotions_tab:
@@ -49,4 +66,16 @@ with goals_tab:
                         for ref in r["reference"]:
                             st.markdown(f"> {ref}")
 
-                        st.write("\n")
+                        st.write("\n")
+
+with st.sidebar:
+    for ch in reports.keys():
+        container = st.container(border=True)
+        col1, col2 = container.columns([2, 1])
+
+        with col1:
+            st.header(ch)
+
+        with col2:
+            st.button("Show", key=ch, on_click=lambda ch=ch: st.session_state.update(character=ch))
+

hp_emotions.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f3615d03acc4e804f3867ee464d16ff73a979a3caeb6b05663718cb304f8bcb9
+size 198744

hp_goals.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:48f7d81cd39a3d3ecff90caeec691c22a94c39ebf3c24db5ed3e79857df8414d
+size 125473