app.py

import streamlit as st
from transformers import pipeline
from nltk.tokenize import sent_tokenize
import base64
import torch
import nltk
import os
import tempfile


@st.cache_resource
def init_nltk():
    # ✅ Use a writable temp directory, not /app
    nltk_data_dir = os.path.join(tempfile.gettempdir(), "nltk_data")
    os.makedirs(nltk_data_dir, exist_ok=True)

    # ✅ Register this directory for nltk
    if nltk_data_dir not in nltk.data.path:
        nltk.data.path.insert(0, nltk_data_dir)

    # ✅ Download only required packages
    for pkg in ["punkt"]:
        try:
            nltk.data.find(f"tokenizers/{pkg}")
        except LookupError:
            nltk.download(pkg, download_dir=nltk_data_dir, quiet=True)

    return True

# ✅ Run it once at startup
init_nltk()



# Device detection
DEVICE = 0 if torch.cuda.is_available() else -1

# Lazy model loading
def get_summarizer():
    if 'summarizer' not in st.session_state:
        with st.spinner("🧠 Loading AI Brain..."):
            st.session_state.summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", device=DEVICE)
    return st.session_state.summarizer

def get_qa():
    if 'qa' not in st.session_state:
        with st.spinner("🧠 Loading AI Brain..."):
            st.session_state.qa = pipeline("question-answering", model="distilbert-base-uncased-distilled-squad", device=DEVICE)
    return st.session_state.qa

def get_classifier():
    if 'classifier' not in st.session_state:
        with st.spinner("🧠 Loading AI Brain..."):
            st.session_state.classifier = pipeline("zero-shot-classification", model="typeform/distilbert-base-uncased-mnli", device=DEVICE)
    return st.session_state.classifier

@st.cache_resource
def load_translator(model_name):
    return pipeline("translation", model=model_name, device=DEVICE)

def truncate_text(text, max_words=400):
    words = text.split()
    return (" ".join(words[:max_words]), len(words) > max_words)

# ULTRA PREMIUM CSS - Glassmorphism + Animations - FIXED HEADER
st.markdown("""

<style>

@import url('https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;600;700;900&display=swap');

* {

    font-family: 'Poppins', sans-serif;

}

.stApp {

    background: linear-gradient(135deg, #667eea 0%, #764ba2 25%, #f093fb 50%, #4facfe 75%, #00f2fe 100%);

    background-size: 400% 400%;

    animation: gradientShift 15s ease infinite;

}

@keyframes gradientShift {

    0% { background-position: 0% 50%; }

    50% { background-position: 100% 50%; }

    100% { background-position: 0% 50%; }

}

/* NEW HEADER - Simple and Visible */

.hero-header {

    background: linear-gradient(135deg, #1e1e3f 0%, #2d2d5f 100%);

    padding: 3rem 2rem;

    border-radius: 25px;

    margin-bottom: 2rem;

    text-align: center;

    box-shadow: 0 20px 60px rgba(0, 0, 0, 0.5);

    border: 2px solid #667eea;

}



.hero-title {

    font-size: 3.5rem;

    font-weight: 900;

    color: #ffffff;

    margin: 0 0 1rem 0;

    text-shadow: 2px 2px 8px rgba(0, 0, 0, 0.5);

}



.hero-subtitle {

    font-size: 1.3rem;

    color: #ffffff;

    margin: 0;

    font-weight: 400;

    opacity: 0.95;

}

/* Premium Tabs */

.stTabs [data-baseweb="tab-list"] {

    gap: 12px;

    background: rgba(255, 255, 255, 0.1);

    padding: 12px;

    border-radius: 20px;

    backdrop-filter: blur(10px);

    border: 1px solid rgba(255, 255, 255, 0.2);

}

.stTabs [data-baseweb="tab"] {

    background: rgba(255, 255, 255, 0.15);

    border-radius: 15px;

    color: white;

    font-weight: 600;

    font-size: 1.1rem;

    padding: 12px 24px;

    border: 1px solid rgba(255, 255, 255, 0.25);

    transition: all 0.3s ease;

    text-shadow: 0 2px 4px rgba(0, 0, 0, 0.2);

}

.stTabs [data-baseweb="tab"]:hover {

    background: rgba(255, 255, 255, 0.25);

    transform: translateY(-2px);

    box-shadow: 0 4px 12px rgba(0, 0, 0, 0.2);

}

.stTabs [aria-selected="true"] {

    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);

    border-color: rgba(255, 255, 255, 0.4);

    box-shadow: 0 8px 24px rgba(102, 126, 234, 0.4);

}

/* Premium Buttons */

.stButton > button {

    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);

    color: white;

    font-weight: 700;

    font-size: 1.1rem;

    padding: 16px 40px;

    border-radius: 16px;

    border: none;

    box-shadow: 0 8px 24px rgba(102, 126, 234, 0.4);

    transition: all 0.3s ease;

    width: 100%;

    text-shadow: 0 2px 4px rgba(0, 0, 0, 0.2);

}

.stButton > button:hover {

    transform: translateY(-3px);

    box-shadow: 0 12px 36px rgba(102, 126, 234, 0.6);

    background: linear-gradient(135deg, #764ba2 0%, #667eea 100%);

}

/* Input Fields */

.stTextArea textarea, .stTextInput input {

    background: rgba(255, 255, 255, 0.2) !important;

    backdrop-filter: blur(10px);

    border: 2px solid rgba(255, 255, 255, 0.3) !important;

    border-radius: 16px !important;

    color: white !important;

    font-size: 1rem !important;

    padding: 16px !important;

    text-shadow: 0 1px 2px rgba(0, 0, 0, 0.2);

}

.stTextArea textarea::placeholder, .stTextInput input::placeholder {

    color: rgba(255, 255, 255, 0.7) !important;

}

.stTextArea textarea:focus, .stTextInput input:focus {

    border-color: rgba(255, 255, 255, 0.6) !important;

    box-shadow: 0 0 20px rgba(255, 255, 255, 0.3) !important;

    background: rgba(255, 255, 255, 0.25) !important;

}

/* Result Cards */

.result-card {

    background: rgba(255, 255, 255, 0.95);

    color: #1a1a1a;

    padding: 2rem;

    border-radius: 20px;

    margin: 1rem 0;

    box-shadow: 0 8px 32px rgba(0, 0, 0, 0.2);

    animation: fadeIn 0.5s ease;

    border-left: 5px solid #667eea;

}

.result-card p {

    word-break: break-word;

    overflow-wrap: break-word;

    max-height: 300px;

    overflow-y: auto;

}

@keyframes fadeIn {

    from { opacity: 0; transform: translateY(20px); }

    to { opacity: 1; transform: translateY(0); }

}

/* Stats Badge */

.stats-badge {

    display: inline-block;

    background: rgba(255, 255, 255, 0.25);

    backdrop-filter: blur(10px);

    padding: 8px 20px;

    border-radius: 20px;

    color: white;

    font-weight: 600;

    border: 1px solid rgba(255, 255, 255, 0.3);

    margin: 5px;

    text-shadow: 0 1px 3px rgba(0, 0, 0, 0.3);

}

/* Success/Error Messages */

.stSuccess {

    background: rgba(72, 187, 120, 0.25) !important;

    backdrop-filter: blur(10px);

    border-left: 4px solid #48bb78 !important;

    border-radius: 12px !important;

    color: white !important;

}

.stError {

    background: rgba(245, 101, 101, 0.25) !important;

    backdrop-filter: blur(10px);

    border-left: 4px solid #f56565 !important;

    border-radius: 12px !important;

    color: white !important;

}

.stInfo {

    background: rgba(66, 153, 225, 0.25) !important;

    backdrop-filter: blur(10px);

    border-left: 4px solid #4299e1 !important;

    border-radius: 12px !important;

    color: white !important;

}

/* Sidebar */

.css-1d391kg, [data-testid="stSidebar"] {

    background: rgba(20, 20, 40, 0.9);

    backdrop-filter: blur(20px);

    border-right: 1px solid rgba(255, 255, 255, 0.2);

}

/* Ensure text wraps properly */

.stTabs h3 {

    word-break: break-word;

    overflow-wrap: anywhere;

    white-space: normal;

    max-width: 100%;

    margin: 0;

    padding: 0.5rem 0;

    color: white;

    text-shadow: 0 2px 4px rgba(0, 0, 0, 0.3);

}

.css-1d391kg h2, [data-testid="stSidebar"] h2 {

    color: white;

    font-weight: 700;

    text-shadow: 0 2px 4px rgba(0, 0, 0, 0.3);

}

.css-1d391kg p, [data-testid="stSidebar"] p {

    color: rgba(255, 255, 255, 0.9);

}

/* Spinner */

.stSpinner > div {

    border-top-color: white !important;

}

/* Selectbox */

.stSelectbox > div > div {

    background: rgba(255, 255, 255, 0.2) !important;

    backdrop-filter: blur(10px);

    border-radius: 12px !important;

    color: white !important;

    border: 2px solid rgba(255, 255, 255, 0.3) !important;

}

.stSelectbox label {

    color: white !important;

    font-weight: 600 !important;

    text-shadow: 0 1px 3px rgba(0, 0, 0, 0.3);

}

/* Footer */

.premium-footer {

    text-align: center;

    padding: 2rem;

    margin-top: 3rem;

    background: rgba(20, 20, 40, 0.85);

    backdrop-filter: blur(15px);

    border-radius: 20px;

    border: 2px solid rgba(255, 255, 255, 0.3);

    box-shadow: 0 8px 32px rgba(0, 0, 0, 0.3);

}

.premium-footer p {

    color: rgba(255, 255, 255, 0.9);

    margin: 0.5rem 0;

    text-shadow: 0 1px 3px rgba(0, 0, 0, 0.3);

}

/* Hide Streamlit Branding */

#MainMenu {visibility: hidden;}

footer {visibility: hidden;}

header {visibility: hidden;}



/* Custom scrollbar */

::-webkit-scrollbar {

    width: 10px;

}

::-webkit-scrollbar-track {

    background: rgba(255, 255, 255, 0.1);

}

::-webkit-scrollbar-thumb {

    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);

    border-radius: 5px;

}

::-webkit-scrollbar-thumb:hover {

    background: linear-gradient(135deg, #764ba2 0%, #667eea 100%);

}

</style>

""", unsafe_allow_html=True)

# New Simple Header
st.markdown("""

<div class="hero-header">

    <div class="hero-title">🧠 AI Study Helper Pro</div>

    <div class="hero-subtitle">⚡ Supercharge Your Learning with Advanced AI Technology</div>

</div>

""", unsafe_allow_html=True)

# Sidebar
with st.sidebar:
    st.markdown("### 🎯 Dashboard")
    st.markdown("---")
    
    # Stats
    col1, col2 = st.columns(2)
    with col1:
        st.markdown('<div class="stats-badge">📊 247 Processed</div>', unsafe_allow_html=True)
    with col2:
        st.markdown('<div class="stats-badge">⚡ 2.3s Avg</div>', unsafe_allow_html=True)
    
    st.markdown("---")
    st.markdown("### ✨ Features")
    features = [
        "📝 AI Summarization",
        "💬 Smart Q&A",
        "🎯 Quiz Generator",
        "🌍 Multi-Language",
        "🔑 Keyword Extraction",
        "💨 Lightning Fast"
    ]
    for feat in features:
        st.markdown(f"**{feat}**")
    
    st.markdown("---")
    st.markdown("### 👩‍💻 Developer")
    st.markdown("**Umaima Qureshi**")
    st.markdown("[GitHub](https://github.com/Umaima122)")

# Session state
for key in ["summary", "quiz", "translation", "keywords"]:
    if key not in st.session_state:
        st.session_state[key] = "" if key not in ["quiz", "keywords"] else []

# Tabs
tab1, tab2, tab3, tab4, tab5, tab6 = st.tabs([
    "📝 Summarize", "💬 Q&A", "🎯 Quiz", "🌍 Translate", "🔑 Keywords", "📥 Download"
])

# SUMMARIZE TAB
with tab1:
    st.markdown("### 📝 Intelligent Summarization")
    
    text = st.text_area("✍️ Your notes or textbook:", value="", height=250, key="sum_txt", 
                        placeholder="Paste your content here and watch AI magic happen...")
    
    col1, col2, col3 = st.columns([1, 1, 1])
    with col2:
        if st.button("✨ Generate Summary", key="sum_btn"):
            if not text.strip():
                st.error("⚠️ Please provide text to summarize")
            else:
                trunc, was_trunc = truncate_text(text, 400)
                if was_trunc:
                    st.info("📊 Text optimized to 400 words")
                
                if len(trunc.split()) < 20:
                    st.error("⚠️ Need at least 20 words")
                else:
                    with st.spinner("🧠 AI is thinking..."):
                        try:
                            summarizer = get_summarizer()
                            result = summarizer(trunc, max_length=130, min_length=30, do_sample=False, truncation=True)
                            summary = result[0]['summary_text']
                            
                            st.markdown(f"""

                            <div class="result-card">

                                <h4 style="color: #667eea; margin-bottom: 1rem;">📄 AI-Generated Summary</h4>

                                <p style="font-size: 1.1rem; line-height: 1.8; color: #2d3748;">{summary}</p>

                                <div style="margin-top: 1rem; padding-top: 1rem; border-top: 1px solid #e2e8f0;">

                                    <span class="stats-badge" style="background: #667eea; color: white;">

                                        {len(summary.split())} words

                                    </span>

                                    <span class="stats-badge" style="background: #48bb78; color: white;">

                                        ✓ Completed

                                    </span>

                                </div>

                            </div>

                            """, unsafe_allow_html=True)
                            st.session_state["summary"] = summary
                        except Exception as e:
                            st.error(f"❌ Error: {str(e)}")

# Q&A TAB
with tab2:
    st.markdown("### 💬 Intelligent Q&A System")
    
    context = st.text_area("📚 Context (Your notes):", value="", height=200, key="qa_ctx",
                          placeholder="Paste your study material here...")
    question = st.text_input("❓ Ask your question:", key="qa_q",
                            placeholder="What would you like to know?")
    
    col1, col2, col3 = st.columns([1, 1, 1])
    with col2:
        if st.button("🔍 Get Answer", key="qa_btn"):
            if not context.strip() or not question.strip():
                st.error("⚠️ Please provide both context and question")
            else:
                trunc_ctx, _ = truncate_text(context, 400)
                with st.spinner("🤔 Analyzing..."):
                    try:
                        qa_model = get_qa()
                        answer = qa_model(question=question, context=trunc_ctx)['answer']
                        
                        st.markdown(f"""

                        <div class="result-card">

                            <h4 style="color: #667eea; margin-bottom: 1rem;">💡 AI Answer</h4>

                            <p style="font-size: 1.2rem; line-height: 1.8; color: #2d3748; font-weight: 500;">{answer}</p>

                            <div style="margin-top: 1rem; padding-top: 1rem; border-top: 1px solid #e2e8f0;">

                                <span class="stats-badge" style="background: #48bb78; color: white;">

                                    ✓ Answer Found

                                </span>

                            </div>

                        </div>

                        """, unsafe_allow_html=True)
                    except Exception as e:
                        st.error(f"❌ Error: {str(e)}")

# QUIZ TAB
with tab3:
    st.markdown("### 🎯 AI Quiz Generator")
    
    quiz_ctx = st.text_area("📖 Study material:", value="", height=200, key="quiz_ctx",
                           placeholder="Paste content for quiz generation...")
    
    col1, col2, col3 = st.columns([1, 1, 1])
    with col2:
        if st.button("🚀 Generate Quiz", key="quiz_btn"):
            if not quiz_ctx.strip():
                st.error("⚠️ Please provide text")
            else:
                trunc_quiz, _ = truncate_text(quiz_ctx, 200)
                with st.spinner("🎲 Creating questions..."):
                    sentences = sent_tokenize(trunc_quiz)[:5]
                    questions = [f"What is the main concept in: '{s[:70]}...'?" for s in sentences]
                    
                    st.markdown('<div class="result-card">', unsafe_allow_html=True)
                    st.markdown("<h4 style='color: #667eea;'>📝 Generated Quiz Questions</h4>", unsafe_allow_html=True)
                    for i, q in enumerate(questions, 1):
                        st.markdown(f"""

                        <div style='background: #f7fafc; padding: 1rem; margin: 0.5rem 0; border-radius: 12px; border-left: 4px solid #667eea;'>

                            <strong style='color: #667eea;'>Question {i}:</strong> {q}

                        </div>

                        """, unsafe_allow_html=True)
                    st.markdown('</div>', unsafe_allow_html=True)
                    st.session_state["quiz"] = questions

# TRANSLATE TAB
with tab4:
    st.markdown("### 🌍 AI Translation")
    
    trans_text = st.text_area("✍️ Text to translate:", height=200, key="trans_txt",
                              placeholder="Enter text to translate...")
    
    col1, col2 = st.columns(2)
    with col1:
        lang = st.selectbox("🎯 Target language:", ["French", "German", "Spanish", "Italian", "Hindi"])
    
    with col2:
        st.write("")
        st.write("")
        if st.button("🌐 Translate Now", key="trans_btn"):
            if not trans_text.strip():
                st.error("⚠️ Please provide text")
            else:
                model_map = {
                    "French": "Helsinki-NLP/opus-mt-en-fr",
                    "German": "Helsinki-NLP/opus-mt-en-de",
                    "Spanish": "Helsinki-NLP/opus-mt-en-es",
                    "Italian": "Helsinki-NLP/opus-mt-en-it",
                    "Hindi": "Helsinki-NLP/opus-mt-en-hi"
                }
                
                trunc_trans, _ = truncate_text(trans_text, 200)
                with st.spinner(f"🌍 Translating to {lang}..."):
                    try:
                        translator = load_translator(model_map[lang])
                        translation = translator(trunc_trans, max_length=256)[0]['translation_text']
                        
                        st.markdown(f"""

                        <div class="result-card">

                            <h4 style="color: #667eea; margin-bottom: 1rem;">🌐 Translation ({lang})</h4>

                            <p style="font-size: 1.2rem; line-height: 1.8; color: #2d3748;">{translation}</p>

                            <div style="margin-top: 1rem; padding-top: 1rem; border-top: 1px solid #e2e8f0;">

                                <span class="stats-badge" style="background: #48bb78; color: white;">

                                    ✓ Translated

                                </span>

                            </div>

                        </div>

                        """, unsafe_allow_html=True)
                        st.session_state["translation"] = translation
                    except Exception as e:
                        st.error(f"❌ Error: {str(e)}")

# KEYWORDS TAB
with tab5:
    st.markdown("### 🔑 AI Keyword Extraction")
    
    keyword_input = st.text_area("📝 Text for analysis:", value="", height=200, key="kw_txt",
                                 placeholder="Paste text to extract key concepts...")
    
    col1, col2, col3 = st.columns([1, 1, 1])
    with col2:
        if st.button("🔍 Extract Keywords", key="kw_btn"):
            if not keyword_input.strip():
                st.error("⚠️ Please provide text")
            else:
                trunc_kw, _ = truncate_text(keyword_input, 200)
                with st.spinner("🔎 Analyzing concepts..."):
                    try:
                        classifier = get_classifier()
                        labels = ["technology", "science", "education", "health", "business", "finance", "medical"]
                        result = classifier(trunc_kw, labels)
                        keywords = [lbl for lbl, score in zip(result['labels'], result['scores']) if score > 0.3][:5]
                        
                        if keywords:
                            st.markdown('<div class="result-card">', unsafe_allow_html=True)
                            st.markdown("<h4 style='color: #667eea;'>🎯 Extracted Keywords</h4>", unsafe_allow_html=True)
                            kw_html = " ".join([
                                f"<span style='display: inline-block; background: linear-gradient(135deg, #667eea, #764ba2); color: white; padding: 12px 24px; border-radius: 25px; margin: 8px; font-size: 1rem; font-weight: 600; box-shadow: 0 4px 12px rgba(102, 126, 234, 0.3);'>{kw}</span>"
                                for kw in keywords
                            ])
                            st.markdown(kw_html, unsafe_allow_html=True)
                            st.markdown('</div>', unsafe_allow_html=True)
                            st.session_state["keywords"] = keywords
                        else:
                            st.info("ℹ️ No strong keywords found")
                    except Exception as e:
                        st.error(f"❌ Error: {str(e)}")

# DOWNLOAD TAB
with tab6:
    st.markdown("### 📥 Download Results")
    
    def download_link(text, filename, emoji):
        b64 = base64.b64encode(text.encode()).decode()
        return f"""

        <a href="data:file/txt;base64,{b64}" download="{filename}" 

           style="display: inline-block; background: linear-gradient(135deg, #667eea, #764ba2); 

           color: white; padding: 16px 32px; border-radius: 16px; text-decoration: none; 

           font-weight: 700; font-size: 1.1rem; margin: 10px; box-shadow: 0 8px 24px rgba(102, 126, 234, 0.3);

           transition: all 0.3s ease;">

            {emoji} Download {filename}

        </a>

        """
    
    col1, col2 = st.columns(2)
    
    with col1:
        if st.session_state["summary"]:
            st.markdown(download_link(st.session_state["summary"], "summary.txt", "📄"), unsafe_allow_html=True)
        if st.session_state["quiz"]:
            st.markdown(download_link("\n".join(st.session_state["quiz"]), "quiz.txt", "🎯"), unsafe_allow_html=True)
    
    with col2:
        if st.session_state["translation"]:
            st.markdown(download_link(st.session_state["translation"], "translation.txt", "🌍"), unsafe_allow_html=True)
        if st.session_state["keywords"]:
            st.markdown(download_link(", ".join(st.session_state["keywords"]), "keywords.txt", "🔑"), unsafe_allow_html=True)
    
    if not any([st.session_state["summary"], st.session_state["quiz"], 
                st.session_state["translation"], st.session_state["keywords"]]):
        st.info("ℹ️ Generate content in other tabs to enable downloads")

# Premium Footer
st.markdown("""

<div class="premium-footer">

    <p style="font-size: 1.2rem; font-weight: 600;">Built with ❤️ by <span style="background: linear-gradient(135deg, #ffffff, #e0e7ff); -webkit-background-clip: text; -webkit-text-fill-color: transparent; font-weight: 700;">Umaima Qureshi</span></p>

    <p style="font-size: 0.9rem;">© 2025 AI Study Helper Pro. All Rights Reserved.</p>

    <p style="margin-top: 1rem;">

        <a href="https://github.com/Umaima122" target="_blank" 

           style="color: white; text-decoration: none; padding: 8px 20px; background: rgba(255, 255, 255, 0.15); 

           border-radius: 20px; backdrop-filter: blur(10px); border: 1px solid rgba(255, 255, 255, 0.3);

           transition: all 0.3s ease;">

            🔗 GitHub

        </a>

    </p>

</div>

""", unsafe_allow_html=True)