Update app.py

app.py

@@ -7,12 +7,16 @@ import pandas as pd
 
 def main():
     st.title("WhatsApp Analysis Tool")
+    st.markdown("This app summarizes Whatsapp chats and provides named entity recognition as well as sentiment analysis for the conversation")
+    st.markdown("**NOTE**: *This app can only receive chats downloaded from IOS as the downloaded chat format is different than from Android.*")
+    st.markdown("Download your whatsapp chat by going to Settings > Chats > Export Chat and there select the chat you want to summarize (download 'Without Media').")
+
 
     # File uploader
-    uploaded_file = st.file_uploader("Choose a file (.txt or .zip)", type=['txt', 'zip'])
+    uploaded_file = st.file_uploader("Choose a file (.zip)", type=['zip'])
     if uploaded_file is not None:
         file_type = detect_file_type(uploaded_file.name)
-        if file_type in ["txt", "zip"]:
+        if file_type == "zip":
             # Process the file
             data = preprocess_whatsapp_messages(uploaded_file, file_type)
             if data.empty:
@@ -26,6 +30,12 @@ def main():
                     text_for_analysis = get_dated_input(data, selected_date)
                     with st.expander("Show/Hide Original Conversation"):
                         st.markdown(f"```\n{text_for_analysis}\n```", unsafe_allow_html=True)
+                    process = st.button('Process')
+                    if process:
+                        # Load models
+                        tokenizer_sentiment, model_sentiment = load_sentiment_analyzer()
+                        tokenizer_summary, model_summary = load_summarizer()
+                        pipe_ner = load_NER()
 
                     # Load models
                     tokenizer_sentiment, model_sentiment = load_sentiment_analyzer()