Update app.py

app.py

@@ -1,4 +1,4 @@
-# # app.py
+
 
 # import os
 # import re
@@ -13,7 +13,7 @@
 
 # # ✅ Hugging Face and GROQ secrets loaded via Hugging Face Spaces Secrets interface
 
-# # ⛳ Access secrets securely from environment variables
+# # ⚪ Access secrets securely from environment variables
 # GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 # HF_TOKEN = os.getenv("HF_TOKEN")
 # KAGGLE_USERNAME = os.getenv("KAGGLE_USERNAME")
@@ -59,6 +59,8 @@
 #     st.session_state.history = []
 # if "audio_summary" not in st.session_state:
 #     st.session_state.audio_summary = ""
+# if "report_sent" not in st.session_state:
+#     st.session_state.report_sent = False
 
 # # =======================
 # # Streamlit UI
@@ -173,9 +175,11 @@
 # if clear_btn:
 #     st.session_state.history.clear()
 #     st.session_state.audio_summary = ""
+#     st.session_state.report_sent = False
 #     st.success("✅ History cleared!")
 
 # if analyze_btn:
+#     st.session_state.report_sent = False
 #     combined_text = text_input
 #     if uploaded_file:
 #         extracted = extract_text_from_file(uploaded_file)
@@ -207,9 +211,6 @@
 #                         st.audio(f.read(), format="audio/mp3")
 #                     os.remove(audio_path)
 
-#                 if st.button("📤 Send Report to IT"):
-#                     st.success("📨 Report sent to IT successfully.")
-
 #         # Save history
 #         st.session_state.history.append({
 #             "input": combined_text,
@@ -230,10 +231,19 @@
 #             for term, definition in GLOSSARY.items():
 #                 st.markdown(f"**{term.capitalize()}**: {definition}")
 
+# # ✅ Report to IT section - outside the expander and stable
+# st.markdown("---")
+# report_it = st.button("📤 Report to IT", key="report_it_btn")
+# if report_it:
+#     st.session_state.report_sent = True
+
+# if st.session_state.report_sent:
+#     st.success("📨 Report sent to IT successfully.")
+
 # render_history()
 
 
-# app.py
+
 # app.py
 
 import os
@@ -297,6 +307,12 @@ if "audio_summary" not in st.session_state:
     st.session_state.audio_summary = ""
 if "report_sent" not in st.session_state:
     st.session_state.report_sent = False
+if "chat_active" not in st.session_state:
+    st.session_state.chat_active = False
+if "text_input" not in st.session_state:
+    st.session_state.text_input = ""
+if "uploaded_file" not in st.session_state:
+    st.session_state.uploaded_file = None
 
 # =======================
 # Streamlit UI
@@ -306,11 +322,20 @@ st.set_page_config(page_title="ZeroPhish Gate", layout="wide")
 st.title("🛡️ ZeroPhish Gate")
 st.markdown("AI-powered phishing message detection and explanation.")
 
+# ✅ New Chat button
+if st.button("🆕 New Chat"):
+    st.session_state.chat_active = False
+    st.session_state.audio_summary = ""
+    st.session_state.report_sent = False
+    st.session_state.text_input = ""
+    st.session_state.uploaded_file = None
+    st.experimental_rerun()
+
 # Input fields
 col1, col2 = st.columns([3, 1])
 with col1:
-    text_input = st.text_area("✉️ Paste Suspicious Message", height=200)
-    uploaded_file = st.file_uploader("📄 Upload PDF/TXT (optional)", type=["pdf", "txt"])
+    st.session_state.text_input = st.text_area("✉️ Paste Suspicious Message", value=st.session_state.text_input, height=200)
+    st.session_state.uploaded_file = st.file_uploader("📄 Upload PDF/TXT (optional)", type=["pdf", "txt"])
 
 with col2:
     language = st.selectbox("🌐 Preferred Language", language_choices)
@@ -416,9 +441,10 @@ if clear_btn:
 
 if analyze_btn:
     st.session_state.report_sent = False
-    combined_text = text_input
-    if uploaded_file:
-        extracted = extract_text_from_file(uploaded_file)
+    st.session_state.chat_active = True
+    combined_text = st.session_state.text_input
+    if st.session_state.uploaded_file:
+        extracted = extract_text_from_file(st.session_state.uploaded_file)
         combined_text += "\n" + extracted
 
     if not combined_text.strip():
@@ -467,14 +493,15 @@ if analyze_btn:
             for term, definition in GLOSSARY.items():
                 st.markdown(f"**{term.capitalize()}**: {definition}")
 
-# ✅ Report to IT section - outside the expander and stable
-st.markdown("---")
-report_it = st.button("📤 Report to IT", key="report_it_btn")
-if report_it:
-    st.session_state.report_sent = True
+# ✅ Report to IT section - only visible after analysis
+if st.session_state.chat_active:
+    st.markdown("---")
+    report_it = st.button("📤 Report to IT", key="report_it_btn")
+    if report_it:
+        st.session_state.report_sent = True
 
-if st.session_state.report_sent:
-    st.success("📨 Report sent to IT successfully.")
+    if st.session_state.report_sent:
+        st.success("📨 Report sent to IT successfully.")
 
 render_history()