Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,143 +1,201 @@
+
 import streamlit as st
 import feedparser
 import requests
-from bs4 import BeautifulSoup
-from newspaper import Article
-import time
 import json
-import datetime
-import hashlib
 import os
+from bs4 import BeautifulSoup
+from transformers import MarianMTModel, MarianTokenizer
+import whisper
+from streamlit_audio_recorder import st_audiorec
 
-# --- Page Config ---
-st.set_page_config(
-    page_title="భారత్ పల్స్ (BharatPulse)",
-    page_icon="📰",
-    layout="wide"
-)
-
-# --- Auth Setup ---
-USERS_FILE = "/tmp/users.json"
+st.set_page_config(page_title="📍 BharatPulse - Local News & Sentiment", layout="wide")
 
-def hash_password(password):
-    return hashlib.sha256(password.encode()).hexdigest()
+# ------------------------------
+# User Auth Functions
+# ------------------------------
+USERS_FILE = "users.json"
 
 def load_users():
-    if os.path.exists(USERS_FILE):
-        with open(USERS_FILE, "r") as f:
-            return json.load(f)
-    return {}
+    if not os.path.exists(USERS_FILE):
+        return {}
+    with open(USERS_FILE, "r") as f:
+        return json.load(f)
 
 def save_users(users):
     with open(USERS_FILE, "w") as f:
         json.dump(users, f, indent=2)
 
-def login_user(username, password):
+def register_user(username, email, password, location):
     users = load_users()
-    if username in users and users[username]["password"] == hash_password(password):
-        return users[username]
-    return None
-
-def register_user(username, email, password, location=""):
-    users = load_users()
-    if username in users:
-        return False, "User already exists"
-    users[username] = {
+    if email in users:
+        return False, "User already exists."
+    users[email] = {
+        "username": username,
         "email": email,
-        "password": hash_password(password),
-        "location": location,
-        "created": datetime.datetime.now().strftime("%Y-%m-%d")
+        "password": password,
+        "location": location
     }
     save_users(users)
-    return True, "Registration successful"
+    return True, "Registration successful."
 
-# --- Session Init ---
+def login_user(email, password):
+    users = load_users()
+    user = users.get(email)
+    if user and user["password"] == password:
+        return True, user
+    return False, "Invalid credentials."
+
+# ------------------------------
+# Whisper Model
+# ------------------------------
+@st.cache_resource
+def load_whisper_model():
+    return whisper.load_model("base")
+
+def transcribe_audio(audio_bytes):
+    path = "/tmp/voice.wav"
+    with open(path, "wb") as f:
+        f.write(audio_bytes)
+    model = load_whisper_model()
+    result = model.transcribe(path, language="te")
+    return result["text"]
+
+# ------------------------------
+# Translation Model
+# ------------------------------
+@st.cache_resource
+def load_translation_model():
+    model_name = "Helsinki-NLP/opus-mt-en-te"
+    tokenizer = MarianTokenizer.from_pretrained(model_name)
+    model = MarianMTModel.from_pretrained(model_name)
+    return tokenizer, model
+
+def translate_to_telugu(text):
+    tokenizer, model = load_translation_model()
+    tokens = tokenizer.prepare_seq2seq_batch([text], return_tensors="pt", truncation=True, max_length=512)
+    translated = model.generate(**tokens)
+    return tokenizer.decode(translated[0], skip_special_tokens=True)
+
+# ------------------------------
+# Extract Article Summary
+# ------------------------------
+def extract_article_content(url):
+    try:
+        response = requests.get(url, timeout=5)
+        soup = BeautifulSoup(response.text, "html.parser")
+        paragraphs = soup.find_all('p')
+        content = ' '.join([p.get_text() for p in paragraphs[:5]])
+        return content
+    except Exception as e:
+        return str(e)
+
+# ------------------------------
+# Auth UI
+# ------------------------------
 if "logged_in" not in st.session_state:
     st.session_state.logged_in = False
-    st.session_state.username = ""
-
-# --- Login/Register Page ---
-def show_login_page():
-    st.title("🔐 లాగిన్ (Login)")
-    username = st.text_input("వినియోగదారు పేరు (Username)")
-    password = st.text_input("పాస్‌వర్డ్ (Password)", type="password")
-
-    if st.button("లాగిన్ చేయండి (Login)"):
-        user = login_user(username, password)
-        if user:
-            st.session_state.logged_in = True
-            st.session_state.username = username
-            st.session_state.user_email = user.get("email", "")
-            st.session_state.user_location = user.get("location", "")
-            st.rerun()
-        else:
-            st.error("చెల్లని లాగిన్ వివరాలు.")
-
-    with st.expander("👉 ఖాతా రిజిస్టర్ (Register)"):
-        new_username = st.text_input("కొత్త వినియోగదారు పేరు", key="register_username")
-        new_email = st.text_input("ఇమెయిల్", key="register_email")
-        new_password = st.text_input("పాస్‌వర్డ్", type="password", key="register_password")
-        new_location = st.text_input("స్థానం (ఐచ్ఛికం)", key="register_location")
-        if st.button("రిజిస్టర్ (Register)"):
-            success, msg = register_user(new_username, new_email, new_password, new_location)
-            st.success(msg) if success else st.error(msg)
-
-# --- Stop Here if Not Logged In ---
+
 if not st.session_state.logged_in:
-    show_login_page()
+    st.title("🔐 Login to BharatPulse")
+
+    tab1, tab2 = st.tabs(["🔑 Login", "📝 Register"])
+
+    with tab1:
+        email = st.text_input("Email")
+        password = st.text_input("Password", type="password")
+        if st.button("Login"):
+            success, user_or_msg = login_user(email, password)
+            if success:
+                st.session_state.logged_in = True
+                st.session_state.user = user_or_msg
+                st.rerun()
+            else:
+                st.error(user_or_msg)
+
+    with tab2:
+        new_username = st.text_input("Username")
+        new_email = st.text_input("New Email")
+        new_password = st.text_input("New Password", type="password")
+        new_location = st.text_input("Location")
+        if st.button("Register"):
+            success, message = register_user(new_username, new_email, new_password, new_location)
+            if success:
+                st.success(message)
+            else:
+                st.error(message)
     st.stop()
 
-# --- Sidebar & Logout ---
-st.sidebar.title("సెట్టింగ్‌లు")
-if st.sidebar.button("🚪 లాగ్ అవుట్"):
+# ------------------------------
+# Main News UI
+# ------------------------------
+st.sidebar.success(f"Logged in as: {st.session_state.user['username']}")
+if st.sidebar.button("🚪 Logout"):
     st.session_state.logged_in = False
-    st.session_state.username = ""
+    st.session_state.user = None
     st.rerun()
 
-# --- BharatPulse UI ---
-st.title("🇮🇳 భారత్ పల్స్ (BharatPulse)")
-st.subheader(f"హాయ్, {st.session_state.username} 🌟")
-
-main_tabs = st.tabs(["📰 తాజా వార్తలు", "📍 నా ప్రాంతం", "🔍 నగర శోధన", "👤 ప్రొఫైల్"])
-
-# --- 1. Headlines Tab ---
-with main_tabs[0]:
-    st.header("📰 తాజా వార్తలు (Top Headlines)")
-    urls = [
-        "https://www.sakshi.com/rss/district-news/medak.xml",
-        "https://www.sakshi.com/rss/district-news/nizamabad.xml"
-    ]
-    for url in urls:
-        st.subheader(f"📡 Feed: {url.split('/')[-1].replace('.xml','').capitalize()}")
+st.title("📍 BharatPulse")
+tabs = st.tabs(["📰 Headlines", "🔍 Search by City", "🎤 Voice Search"])
+
+feeds = {
+    "Sakshi - Medak": "https://www.sakshi.com/rss/district-news/medak.xml",
+    "Eenadu - AP": "https://www.eenadu.net/rss/andhra-pradesh.xml"
+}
+
+# ------------------------------
+# 📰 Headlines Tab
+# ------------------------------
+with tabs[0]:
+    st.header("📰 Local Telugu News")
+    for name, url in feeds.items():
+        st.subheader(name)
         feed = feedparser.parse(url)
         for entry in feed.entries[:3]:
-            st.markdown(f"### 📰 {entry.title}")
+            st.markdown(f"### {entry.title}")
             st.write(entry.published)
-            st.write(entry.link)
+            st.markdown(f"[Read Full Article]({entry.link})")
+            summary = extract_article_content(entry.link)
+            st.markdown("**English Summary:** " + summary)
+            telugu_summary = translate_to_telugu(summary)
+            st.markdown("**తెలుగు అనువాదం:** " + telugu_summary)
             st.markdown("---")
 
-# --- 2. Location Tab ---
-with main_tabs[1]:
-    st.header("📍 మీ ప్రాంతానికి సంబంధించిన వార్తలు")
-    user_location = st.session_state.get("user_location", "")
-    if user_location:
-        st.write(f"మీ స్థానం: **{user_location}**")
-        st.write("ఇక్కడ మీ స్థానిక వార్తలు చూపించబడతాయి (గత ఇంటిగ్రేషన్ ఆధారంగా).")
-    else:
-        st.info("దయచేసి ప్రొఫైల్‌లో మీ స్థానం జోడించండి.")
-
-# --- 3. City Search Tab ---
-with main_tabs[2]:
-    st.header("🔍 నగర శోధన")
-    city = st.text_input("Search your city/town name:")
+# ------------------------------
+# 🔍 Search by City
+# ------------------------------
+with tabs[1]:
+    st.header("🔎 Search by City / District")
+    city = st.text_input("Enter your city/district name in Telugu or English")
+
     if city:
-        st.write(f"Showing news for: **{city}** (dummy results)")
-        # Later: map city to RSS or Wikidata
-
-# --- 4. Profile Tab ---
-with main_tabs[3]:
-    st.header("👤 మీ ప్రొఫైల్")
-    st.write(f"**Username:** {st.session_state.username}")
-    st.write(f"**Email:** {st.session_state.get('user_email', '')}")
-    st.write(f"**Location:** {st.session_state.get('user_location', '')}")
-    st.write(f"**Account Created On:** {load_users().get(st.session_state.username, {}).get('created', 'NA')}")
+        st.success(f"Showing news for: {city}")
+        for name, url in feeds.items():
+            if city.lower() in name.lower():
+                st.subheader(name)
+                feed = feedparser.parse(url)
+                for entry in feed.entries[:2]:
+                    st.markdown(f"**🗞️ {entry.title}**")
+                    summary = extract_article_content(entry.link)
+                    telugu_summary = translate_to_telugu(summary)
+                    st.markdown("**తెలుగు అనువాదం:** " + telugu_summary)
+
+# ------------------------------
+# 🎤 Voice Search Tab
+# ------------------------------
+with tabs[2]:
+    st.header("🎤 Search by Telugu Voice")
+    audio = st_audiorec()
+    if audio:
+        transcribed = transcribe_audio(audio)
+        st.success(f"Voice Input Recognized: {transcribed}")
+        city = transcribed
+        for name, url in feeds.items():
+            if city.lower() in name.lower():
+                st.subheader(name)
+                feed = feedparser.parse(url)
+                for entry in feed.entries[:2]:
+                    st.markdown(f"**🗞️ {entry.title}**")
+                    summary = extract_article_content(entry.link)
+                    telugu_summary = translate_to_telugu(summary)
+                    st.markdown("**తెలుగు అనువాదం:** " + telugu_summary)