Update frontend.py

frontend.py

@@ -1,4 +1,3 @@
-# --- import statements remain unchanged ---
 import streamlit as st
 import requests
 import pandas as pd
@@ -9,13 +8,14 @@ from PIL import Image
 import os
 import plotly.express as px
 
-# --- Config ---
 st.set_page_config(page_title="NeuroPulse AI", page_icon="🧠", layout="wide")
+
+# Logo
 logo_path = "logo.png"
 if os.path.exists(logo_path):
     st.image(logo_path, width=180)
 
-# --- Session State ---
+# Session State
 if "review" not in st.session_state:
     st.session_state.review = ""
 if "dark_mode" not in st.session_state:
@@ -25,7 +25,7 @@ if "intelligence_mode" not in st.session_state:
 if "trigger_example_analysis" not in st.session_state:
     st.session_state.trigger_example_analysis = False
 
-# --- Dark Mode Styling ---
+# Dark Mode Style
 if st.session_state.dark_mode:
     st.markdown("""
         <style>
@@ -44,11 +44,24 @@ if st.session_state.dark_mode:
         </style>
     """, unsafe_allow_html=True)
 
-# --- Sidebar ---
+# Sidebar Controls
 with st.sidebar:
     st.header("⚙️ Global Settings")
-    st.session_state.dark_mode = st.toggle("🌙 Dark Mode", value=st.session_state.dark_mode)
-    st.session_state.intelligence_mode = st.toggle("🧠 Intelligence Mode", value=st.session_state.intelligence_mode)
+
+    # Dark Mode Toggle with refresh
+    if st.toggle("🌙 Dark Mode", value=st.session_state.dark_mode):
+        if not st.session_state.dark_mode:
+            st.session_state.dark_mode = True
+            st.rerun()
+    else:
+        if st.session_state.dark_mode:
+            st.session_state.dark_mode = False
+            st.rerun()
+
+    if st.toggle("🧠 Intelligence Mode", value=st.session_state.intelligence_mode):
+        st.session_state.intelligence_mode = True
+    else:
+        st.session_state.intelligence_mode = False
 
     DEFAULT_DEMO_TOKEN = "my-secret-key"
     api_token = st.text_input("🔐 API Token", value=DEFAULT_DEMO_TOKEN, type="password")
@@ -80,7 +93,7 @@ with st.sidebar:
     verbosity = st.radio("🗣️ Response Style", ["Brief", "Detailed"])
     voice_lang = st.selectbox("🔈 Voice Language", ["en", "fr", "es", "de", "hi", "zh"])
 
-# --- TTS ---
+# TTS Function
 def speak(text, lang='en'):
     tts = gTTS(text, lang=lang)
     mp3 = BytesIO()
@@ -90,15 +103,13 @@ def speak(text, lang='en'):
     mp3.seek(0)
     return mp3
 
-# --- Tabs ---
 tab1, tab2 = st.tabs(["🧠 Single Review", "📚 Bulk CSV"])
 
-# -------------------
-# SINGLE REVIEW TAB
-# -------------------
+# --- SINGLE REVIEW TAB ---
 with tab1:
     st.title("🧠 NeuroPulse AI – Multimodal Review Analyzer")
     st.markdown("<div style='font-size:16px;color:#888;'>Minimum 20–50 words recommended.</div>", unsafe_allow_html=True)
+
     review = st.text_area("📝 Enter Review", value=st.session_state.review, height=180)
 
     col1, col2, col3 = st.columns(3)
@@ -135,8 +146,7 @@ with tab1:
                         "follow_up": None,
                         "product_category": product_category,
                         "verbosity": verbosity,
-                        "intelligence": st.session_state.intelligence_mode,
-                        "explain": True
+                        "intelligence": st.session_state.intelligence_mode
                     }
                     headers = {"x-api-key": st.session_state.get("api_token", api_token)}
                     params = {"smart": "1"} if use_smart_summary else {}
@@ -157,10 +167,7 @@ with tab1:
                         st.metric("📊 Sentiment", data["sentiment"]["label"], delta=f"{data['sentiment']['score']:.2%}")
                         st.info(f"💢 Emotion: {data['emotion']}")
 
-                        if data.get("explanation"):
-                            st.subheader("🧮 Explanation")
-                            st.markdown(data["explanation"])
-
+                        # Follow-up Section
                         st.markdown("### 🔁 Got questions?")
                         st.info("💬 Ask a follow-up question about this review.")
                         sample_questions = [
@@ -170,7 +177,11 @@ with tab1:
                             "What are the improvement areas?"
                         ]
                         selected_q = st.selectbox("💡 Sample Questions", ["Type your own..."] + sample_questions)
-                        custom_q = st.text_input("🔍 Ask a follow-up", value="" if selected_q == "Type your own..." else selected_q)
+
+                        if selected_q != "Type your own...":
+                            custom_q = selected_q
+                        else:
+                            custom_q = st.text_input("🔍 Ask a follow-up")
 
                         if custom_q:
                             with st.spinner("Thinking..."):