import streamlit as st
from api import get_news, NewsRequest
from utils import text_to_speech
import base64

# Function to auto-play audio
def autoplay_audio(file_path):
    with open(file_path, "rb") as audio_file:
        audio_bytes = audio_file.read()
    b64 = base64.b64encode(audio_bytes).decode()
    audio_tag = f"""
        <audio autoplay>
        <source src="data:audio/mp3;base64,{b64}" type="audio/mp3">
        </audio>
    """
    st.markdown(audio_tag, unsafe_allow_html=True)

# Streamlit UI
def main():
    st.set_page_config(page_title="News Sentiment & Summarization", layout="wide")

    # Sidebar UI
    st.sidebar.markdown(
        "<h1 style='color:#00FFAF; text-align:center;'>📰 News Finder</h1>", 
        unsafe_allow_html=True
    )
    st.sidebar.write("Enter a company name to fetch the latest summarized news with sentiment analysis:")

    company_name = st.sidebar.text_input("🔎 Company Name:")
    fetch_button = st.sidebar.button("Fetch News")
    st.sidebar.markdown("---")
    sentiment_summary_placeholder = st.sidebar.empty()

    st.markdown(
        "<h1 style='text-align:center; color:#00FFAF;'>📢 News Sentiment & Summarization</h1>",
        unsafe_allow_html=True
    )

    if fetch_button and company_name:
        response = get_news(NewsRequest(company=company_name))
        articles = response.get("articles", [])

        if not articles:
            st.warning("No news found for this company.")
            return

        # Sentiment summary display
        sentiments = [article['sentiment'] for article in articles]
        positive = sentiments.count("Positive")
        negative = sentiments.count("Negative")
        neutral = sentiments.count("Neutral")

        sentiment_summary_placeholder.markdown(f"""
            <div style='padding:10px; background-color:#222; border-radius:10px; color:white; text-align:center;'>
                <h3>📊 Sentiment Summary</h3>
                <p>✅ Positive: <b style="color:lightgreen;">{positive}</b></p>
                <p>⚠️ Neutral: <b style="color:orange;">{neutral}</b></p>
                <p>❌ Negative: <b style="color:#FF4B4B;">{negative}</b></p>
            </div>
        """, unsafe_allow_html=True)

        # Auto play audio summary of headlines
        all_headlines = ". ".join([f"{idx + 1}. {article['title']}" for idx, article in enumerate(articles)])
        audio_file = text_to_speech(all_headlines)
        if audio_file:
            autoplay_audio(audio_file)

        st.write("## Latest News Articles")

        # News cards
        for idx, article in enumerate(articles, start=1):
            sentiment_color = {
                "Positive": "#4CAF50",
                "Neutral": "#FFA500",
                "Negative": "#FF4B4B"
            }[article['sentiment']]

            st.markdown(f"""
                <div style='
                    border-radius: 12px; 
                    padding: 20px; 
                    margin-bottom: 20px; 
                    background: linear-gradient(145deg, #1f1f1f, #292929); 
                    box-shadow: rgba(0,0,0,0.4) 0px 5px 15px;
                    color: #f1f1f1; 
                '>
                    <h4 style="color:#00FFAF;">📰 {idx}. {article['title']}</h4>
                    <p><b>📝 Summary:</b> {article['summary']}</p>
                    <p><b>💡 Sentiment:</b> <span style="color:{sentiment_color}; font-weight:bold;">{article['sentiment']}</span></p>
                    <p><b>📅 Published At:</b> {article['published_at']}</p>
                    <a href="{article['link']}" target="_blank" style="color:#00FFAF; text-decoration:none;">🔗 Read Full Article</a>
                </div>
            """, unsafe_allow_html=True)
    elif not fetch_button:
        st.info("Enter a company name in the sidebar and click 'Fetch News' to begin.")

if __name__ == "__main__":
    main()