Spaces:
Running
Running
| import streamlit as st | |
| import requests | |
| import pandas as pd | |
| from gtts import gTTS | |
| import base64 | |
| from io import BytesIO | |
| from PIL import Image | |
| import os | |
| st.set_page_config(page_title="NeuroPulse AI", page_icon="🧠", layout="wide") | |
| logo_path = os.path.join("app", "static", "logo.png") | |
| if os.path.exists(logo_path): | |
| st.image(logo_path, width=160) | |
| # Session state defaults | |
| 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 | |
| # 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) | |
| api_token = st.text_input("🔐 API Token", type="password") | |
| backend_url = st.text_input("🖥️ Backend URL", value="http://0.0.0.0:8000") | |
| sentiment_model = st.selectbox("📊 Sentiment Model", [ | |
| "distilbert-base-uncased-finetuned-sst-2-english", | |
| "nlptown/bert-base-multilingual-uncased-sentiment" | |
| ]) | |
| industry = st.selectbox("🏭 Industry Context", [ | |
| "Auto-detect", "Generic", "E-commerce", "Healthcare", "Education", "Travel", "Banking", "Insurance", "Gaming", "Food Delivery", "Real Estate", "Fitness", "Entertainment" | |
| ]) | |
| product_category = st.selectbox("🧩 Product Category", [ | |
| "Auto-detect", "General", "Mobile Devices", "Laptops", "Healthcare Devices", "Banking App", "Travel Service", "Educational Tool", "Insurance Portal", "Streaming App", "Wearables", "Home Appliances", "Food Apps" | |
| ]) | |
| device_type = st.selectbox("💻 Device Type", [ | |
| "Auto-detect", "Web", "Android", "iOS", "Desktop", "Smartwatch", "Kiosk" | |
| ]) | |
| 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") | |
| 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() | |
| tts.write_to_fp(mp3) | |
| b64 = base64.b64encode(mp3.getvalue()).decode() | |
| st.markdown(f'<audio controls><source src="data:audio/mp3;base64,{b64}" type="audio/mp3"></audio>', unsafe_allow_html=True) | |
| mp3.seek(0) | |
| return mp3 | |
| # Tabs for modes | |
| tab1, tab2 = st.tabs(["🧠 Single Review", "📚 Bulk CSV"]) | |
| # --- SINGLE REVIEW MODE --- | |
| with tab1: | |
| st.title("🧠 NeuroPulse AI – Multimodal Review Analyzer") | |
| st.markdown(""" | |
| <div style='font-size:16px;color:#555;'>Minimum 50–100 words recommended for optimal insights.</div> | |
| """, unsafe_allow_html=True) | |
| review = st.text_area("📝 Enter a Review", value=st.session_state.review, height=180) | |
| col1, col2, col3 = st.columns(3) | |
| with col1: | |
| analyze = st.button("🔍 Analyze", use_container_width=True, disabled=not api_token) | |
| with col2: | |
| if st.button("🎲 Example", use_container_width=True): | |
| st.session_state.review = "App was smooth, but the transaction failed twice on Android during checkout." | |
| st.rerun() | |
| with col3: | |
| if st.button("🧹 Clear", use_container_width=True): | |
| st.session_state.review = "" | |
| st.rerun() | |
| if analyze and review: | |
| if len(review.split()) < 50: | |
| st.error("⚠️ Please enter at least 50 words for meaningful analysis.") | |
| else: | |
| with st.spinner("Analyzing..."): | |
| try: | |
| payload = { | |
| "text": review, | |
| "model": sentiment_model, | |
| "industry": industry, | |
| "aspects": use_aspects, | |
| "follow_up": follow_up, | |
| "product_category": product_category, | |
| "device": device_type | |
| } | |
| headers = {"X-API-Key": api_token} | |
| params = {"smart": "1"} if use_smart_summary else {} | |
| res = requests.post(f"{backend_url}/analyze/", json=payload, headers=headers, params=params) | |
| if res.status_code == 200: | |
| data = res.json() | |
| st.success("✅ Analysis Complete") | |
| st.subheader("📌 Summary") | |
| st.info(data["summary"]) | |
| st.caption(f"🧠 Summary Type: {'Smart Summary' if use_smart_summary else 'Standard Model'}") | |
| st.markdown(f"**Context:** {industry} | {product_category} | {device_type}") | |
| st.subheader("🔊 Audio") | |
| audio = speak(data["summary"], lang=voice_lang) | |
| st.download_button("⬇️ Download Summary Audio", audio.read(), "summary.mp3", mime="audio/mp3") | |
| st.metric("📊 Sentiment", data["sentiment"]["label"], delta=f"{data['sentiment']['score']:.2%}") | |
| st.info(f"💢 Emotion: {data['emotion']}") | |
| if data.get("aspects"): | |
| st.subheader("🔍 Aspects") | |
| for a in data["aspects"]: | |
| st.write(f"🔹 {a['aspect']}: {a['sentiment']} ({a['score']:.2%})") | |
| if data.get("follow_up"): | |
| st.subheader("🧠 Follow-Up Response") | |
| st.warning(data["follow_up"]) | |
| else: | |
| st.error(f"❌ API Error: {res.status_code}") | |
| except Exception as e: | |
| st.error(f"🚫 {e}") | |
| # --- BULK REVIEW MODE --- | |
| with tab2: | |
| st.title("📚 Bulk CSV Upload") | |
| st.markdown(""" | |
| <div style='font-size:16px;'>Upload a CSV with the following columns:<br> | |
| <code>review</code> <span style='color:#aaa;'>(required)</span>, | |
| <code>industry</code>, <code>product_category</code>, <code>device</code> <span style='color:#aaa;'>(optional)</span></div> | |
| """, 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") | |
| uploaded_file = st.file_uploader("📁 Upload your CSV", type="csv") | |
| if uploaded_file and api_token: | |
| try: | |
| df = pd.read_csv(uploaded_file) | |
| if "review" not in df.columns: | |
| st.error("CSV must contain a `review` column.") | |
| else: | |
| st.success(f"✅ Loaded {len(df)} reviews") | |
| for col in ["industry", "product_category", "device"]: | |
| if col not in df.columns: | |
| df[col] = [industry if industry != "Auto-detect" else "Generic"] * len(df) | |
| df[col] = df[col].fillna("").astype(str) | |
| if st.button("📊 Analyze Bulk Reviews", use_container_width=True): | |
| with st.spinner("Processing CSV..."): | |
| try: | |
| payload = { | |
| "reviews": df["review"].tolist(), | |
| "model": sentiment_model, | |
| "aspects": use_aspects, | |
| "industry": df["industry"].tolist(), | |
| "product_category": df["product_category"].tolist(), | |
| "device": df["device"].tolist() | |
| } | |
| headers = {"X-API-Key": api_token} | |
| params = {"smart": "1"} if use_smart_summary_bulk else {} | |
| res = requests.post(f"{backend_url}/bulk/", json=payload, headers=headers, params=params) | |
| if res.status_code == 200: | |
| results = pd.DataFrame(res.json()["results"]) | |
| results["summary_type"] = "Smart" if use_smart_summary_bulk else "Standard" | |
| st.dataframe(results) | |
| 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}") | |
| except Exception as e: | |
| st.error(f"💥 Error: {e}") | |
| except Exception as e: | |
| st.error(f"❌ File Error: {e}") | |