import streamlit as st
import requests
import pandas as pd

# ✅ Replace with your Hugging Face backend URL
# ✅ Use the deployed backend Space URL
BACKEND_URL = "https://manishkumaryadav-news-summarize.hf.space/analyze"



st.title("📊 News Sentiment Analysis & TTS in Hindi")

# Input field for company name
company_name = st.text_input("Enter Company Name", "")

if st.button("Analyze"):
    if not company_name:
        st.warning("⚠️ Please enter a company name.")
    else:
        st.info(f"🔍 Analyzing news for {company_name}...")

        # Send request to Flask backend
        response = requests.post(
            BACKEND_URL,
            json={"company_name": company_name}
        )

        if response.status_code == 200:
            data = response.json()

            st.success("✅ Analysis Complete!")

            # ✅ Display Sentiment Summary
            st.subheader("📊 Sentiment Summary")
            st.json(data["sentiment_summary"])

            # ✅ Display Articles
            st.subheader("📰 Extracted Articles")
            
            df = pd.DataFrame(data["articles"])
            
            for _, article in df.iterrows():
                st.markdown(f"### [{article['title']}]({article['url']})")
                st.write(f"**Summary:** {article['summary']}")
                st.write("---")

            # ✅ Display Hindi TTS Audio
            st.subheader("🔊 Hindi TTS Audio Output")

            audio_file_url = f"{BACKEND_URL}/{data['audio_file']}"
            
            st.audio(audio_file_url, format="audio/mp3")

            st.download_button(
                label="⬇️ Download Hindi TTS Audio",
                data=requests.get(audio_file_url).content,
                file_name=f"{company_name}_TTS.mp3",
                mime="audio/mpeg"
            )
            
        else:
            st.error("❌ Error analyzing news. Please try again.")