Update app.py

app.py

@@ -3,22 +3,22 @@ import torch
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
 import base64
 from datetime import datetime
-
+import streamlit.components.v1 as components
 
 
 party_dict = {
-    'AfD':'./afd_gpt2',
-    'Die Grünen':'./gruene_gpt2',
-    'CDU/CSU':'./cdu_csu_gpt2',
-    'die Linke':'./linke_gpt2', 
+    'AfD': './afd_gpt2',
+    'Die Grünen': './gruene_gpt2',
+    'CDU/CSU': './cdu_csu_gpt2',
+    'die Linke': './linke_gpt2', 
 }
 
 picture_paths = {
-    'blue_check': "./app_pictures/Twitter_Verified_Badge.png",
-    'AfD' : "./app_pictures/afd_logo.png",
-    'CDU/CSU' : "./app_pictures/cdu_logo.png",
-    'Die Grünen' : "./app_pictures/gruene_logo.png",
-    'die Linke' : "./app_pictures/linke_logo.png"
+    'blue_check': "./app_pictures//Twitter_Verified_Badge.png",
+    'AfD': "./app_pictures//afd_logo.png",
+    'CDU/CSU': "./app_pictures//cdu_logo.png",
+    'Die Grünen': "./app_pictures//gruene_logo.png",
+    'die Linke': "./app_pictures//linke_logo.png"
 }
 
 party_info = {
@@ -28,7 +28,22 @@ party_info = {
     "die Linke": ("Die Linke", "@dieLinke")
 }
 
+topic_analysis_screenshots = {
+    'AfD': './topic_analysis//afd_topic_analysis.png',
+    'CDU/CSU': './topic_analysis//cdu_csu_topic_analysis.png',
+    'Die Grünen': './topic_analysis//gruene_topic_analysis.png',
+    'die Linke': './topic_analysis//linke_topic_analysis.png'
+}
 
+def initialize_session_state():
+    if 'show_afd' not in st.session_state:
+        st.session_state['show_afd'] = False
+    if 'show_cdu_csu' not in st.session_state:
+        st.session_state['show_cdu_csu'] = False
+    if 'show_gruene' not in st.session_state:
+        st.session_state['show_gruene'] = False
+    if 'show_linke' not in st.session_state:
+        st.session_state['show_linke'] = False
 
 def generate_tweet(party, prompt):
     device = torch.device("cpu")
@@ -59,7 +74,7 @@ def get_base64_image(image_path):
 
 def main():
     st.title("Tweet GPT")
-    st.sidebar.header("**Analyzing Rhetorical Styles of German Political Parties**")
+    st.sidebar.header("Analyzing Rhetorical Styles of German Political Parties")
     st.sidebar.write("This project analyzes rhetorical differences among major German parties by generating tweets tailored to their communication styles. Our motivation was to understand the nuances in language and messaging strategies, particularly the effective use of social media by right-wing parties to disseminate messages. We aimed to raise awareness of this issue and contribute to a better understanding of varying rhetorical approaches.")
     st.sidebar.markdown("#### Motivation")
     st.sidebar.markdown("- Understand language and messaging nuances")
@@ -67,6 +82,12 @@ def main():
     st.sidebar.markdown("- Provide insights into crafting messages for supporters")
     st.sidebar.markdown("- Raise awareness of right-wing parties' social media use")
     st.sidebar.write("Reference: https://www.zdf.de/nachrichten/politik/deutschland/afd-tiktok-erfolg-strategie-jugendliche-100.html")
+    st.sidebar.write("")
+    st.sidebar.write("")
+    st.sidebar.write("## W&B Training Evaluation Report")
+    st.sidebar.markdown("The W&B reports provide a detailed assessment of our Tweet Generator's training process. They include various metrics and visualizations to analyze model performance, track its progress over time, and identify any weaknesses. ")
+    with st.sidebar.expander("W&B Training Evaluation Report"):
+        components.iframe("https://wandb.ai/ifisch/Tweet_Gen_v2_GPT2/reports/TweetGPT-Training-Evaluation-Process--Vmlldzo4MTU1NjU2?accessToken=4qiq23z5xycyyylaass0xge959em11ldry0deqqp6cr53ugqujfsh3xrzb1r8lsq", height=600, scrolling=True)
 
     tweet = ""
     prompt = ""
@@ -91,16 +112,16 @@ def main():
         current_date = datetime.now().strftime("%B %d, %Y")
         blue_check_base64 = get_base64_image(picture_paths['blue_check'])
         party_logo_base64 = get_base64_image(picture_paths[party])
-        party, username = party_info[party]
+        party_name, username = party_info[party]
 
         if tweet:
             tweet_display = f'''
             <div style="background-color: white; padding: 10px; font-family: Helvetica Neue, sans-serif; border: 1px solid #ccc; color: black; border-radius: 10px;">
                 <div style="display: flex; align-items: center;">
-                    <img src="data:image/png;base64,{party_logo_base64}" alt="{party} Logo" style="width: 40px; height: 40px; border-radius: 50%; margin-right: 10px;">
+                    <img src="data:image/png;base64,{party_logo_base64}" alt="{party_name} Logo" style="width: 40px; height: 40px; border-radius: 50%; margin-right: 10px;">
                     <div style="display: flex; align-items: center;">
                         <div style="display: flex; align-items: center; margin-right: 5px;">
-                             <p style="font-weight: bold; color: black;">{party}<img src="data:image/png;base64,{blue_check_base64}" alt="checkmark" style="width: 20px; height: 20px; vertical-align: middle; margin-left: 2px; margin-right: 2px;"><span style="font-weight: normal; color: gray;"> {username}</span> <span style="font-weight: normal; color: gray;">{current_date}</span></p>
+                             <p style="font-weight: bold; color: black;">{party_name}<img src="data:image/png;base64,{blue_check_base64}" alt="checkmark" style="width: 20px; height: 20px; vertical-align: middle; margin-left: 2px; margin-right: 2px;"><span style="font-weight: normal; color: gray;"> {username}</span> <span style="font-weight: normal; color: gray;">{current_date}</span></p>
                 </div>
                     </div>
                 </div>
@@ -109,9 +130,35 @@ def main():
             '''
             st.markdown(tweet_display, unsafe_allow_html=True)
 
-        st.write("## Our Insights")
-        st.write("We developed a tweet generator that analyzes the dataset for each party. To view the insights, select the button to see the most common topics each party wrote about.")
-        
+    st.write("## Our Insights")
+    st.write("We developed a tweet generator that analyzes the dataset for each party. To view the insights, select the button to see the most common topics each party wrote about.")
+    
+    
+
+   
+
+    col1, col2, col3, col4 = st.columns(4)
+    
+    if col1.button("Die AfD Topic Analysis" ):
+        st.session_state['show_afd'] = not st.session_state['show_afd']
+    if col2.button("CDU/CSU Topic Analysis"):
+        st.session_state['show_cdu_csu'] = not st.session_state['show_cdu_csu']
+    if col3.button("Die Grünen Topic Analysis"):
+        st.session_state['show_gruene'] = not st.session_state['show_gruene']
+    if col4.button("Die Linke Topic Analysis"):
+        st.session_state['show_linke'] = not st.session_state['show_linke']
+
+   
+    if st.session_state['show_afd']:
+        st.image(topic_analysis_screenshots['AfD'], caption='AfD Topic Analysis')
+    if st.session_state['show_cdu_csu']:
+        st.image(topic_analysis_screenshots['CDU/CSU'], caption='CDU/CSU Topic Analysis')
+    if st.session_state['show_gruene']:
+        st.image(topic_analysis_screenshots['Die Grünen'], caption='Die Grünen Topic Analysis')
+    if st.session_state['show_linke']:
+        st.image(topic_analysis_screenshots['die Linke'], caption='die Linke Topic Analysis')
+    
 
 if __name__ == "__main__":
+    initialize_session_state()
     main()