Update frontend.py

frontend.py

@@ -6,6 +6,7 @@ import base64
 from io import BytesIO
 from PIL import Image
 import os
+import plotly.express as px
 
 st.set_page_config(page_title="NeuroPulse AI", page_icon="🧠", layout="wide")
 
@@ -18,11 +19,14 @@ if "review" not in st.session_state:
     st.session_state.review = ""
 if "dark_mode" not in st.session_state:
     st.session_state.dark_mode = False
+if "intelligence_mode" not in st.session_state:
+    st.session_state.intelligence_mode = True
 
 # Shared 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)
 
     api_token = st.text_input("🔐 API Token", type="password")
     backend_url = st.text_input("🖥️ Backend URL", value="http://0.0.0.0:8000")
@@ -47,11 +51,11 @@ with st.sidebar:
     use_aspects = st.checkbox("🔬 Enable Aspect Analysis")
     use_smart_summary = st.checkbox("🧠 Use Smart Summary (Single)")
     use_smart_summary_bulk = st.checkbox("🧠 Smart Summary for Bulk")
+    verbosity = st.radio("🗣️ Response Style", ["Brief", "Detailed"])
     follow_up = st.text_input("🔁 Follow-up Question")
     voice_lang = st.selectbox("🔈 Voice Language", ["en", "fr", "es", "de", "hi", "zh"])
 
 # Text-to-Speech Helper
-
 def speak(text, lang='en'):
     tts = gTTS(text, lang=lang)
     mp3 = BytesIO()
@@ -98,7 +102,9 @@ with tab1:
                         "aspects": use_aspects,
                         "follow_up": follow_up,
                         "product_category": product_category,
-                        "device": device_type
+                        "device": device_type,
+                        "verbosity": verbosity,
+                        "intelligence": st.session_state.intelligence_mode
                     }
                     headers = {"X-API-Key": api_token}
                     params = {"smart": "1"} if use_smart_summary else {}
@@ -122,6 +128,9 @@ with tab1:
                         if data.get("follow_up"):
                             st.subheader("🧠 Follow-Up Response")
                             st.warning(data["follow_up"])
+                        if data.get("explanation"):
+                            st.subheader("🧮 Explain This")
+                            st.markdown(data["explanation"])
                     else:
                         st.error(f"❌ API Error: {res.status_code}")
                 except Exception as e:
@@ -137,8 +146,8 @@ with tab2:
     """, unsafe_allow_html=True)
 
     with st.expander("📄 Sample CSV"):
-    with open("sample_reviews.csv", "rb") as f:
-        st.download_button("⬇️ Download sample CSV", f, file_name="sample_reviews.csv")
+        with open("sample_reviews.csv", "rb") as f:
+            st.download_button("⬇️ Download sample CSV", f, file_name="sample_reviews.csv")
 
     uploaded_file = st.file_uploader("📁 Upload your CSV", type="csv")
 
@@ -164,7 +173,8 @@ with tab2:
                                 "aspects": use_aspects,
                                 "industry": df["industry"].tolist(),
                                 "product_category": df["product_category"].tolist(),
-                                "device": df["device"].tolist()
+                                "device": df["device"].tolist(),
+                                "intelligence": st.session_state.intelligence_mode
                             }
                             headers = {"X-API-Key": api_token}
                             params = {"smart": "1"} if use_smart_summary_bulk else {}
@@ -174,6 +184,9 @@ with tab2:
                                 results = pd.DataFrame(res.json()["results"])
                                 results["summary_type"] = "Smart" if use_smart_summary_bulk else "Standard"
                                 st.dataframe(results)
+                                if "sentiment" in results:
+                                    fig = px.pie(results, names="sentiment", title="Sentiment Distribution")
+                                    st.plotly_chart(fig)
                                 st.download_button("⬇️ Download Results CSV", results.to_csv(index=False), "bulk_results.csv", mime="text/csv")
                             else:
                                 st.error(f"❌ Bulk Analysis Failed: {res.status_code}")