import streamlit as st
from textblob import TextBlob
# The transformers import is here in case you want to integrate a pre-trained model later.
# from transformers import pipeline

def get_roast(user_text):
    """
    Generates a roast based on the user's input.
    (You can replace or extend this logic by integrating a model like Mixtral-8x7B.)
    """
    if "procrastinate" in user_text.lower():
        return "Ah, the art of doing nothing! Do you charge Netflix for your couch imprint? 🛋️"
    elif "always" in user_text.lower() or "never" in user_text.lower():
        return "Wow, painting your world in extremes? Maybe it's time to add some shades of gray!"
    else:
        return "Is that a problem or a lifestyle choice? Time to get serious... or maybe not."

def analyze_text(user_text):
    """
    Analyzes the input text for sentiment and detects basic cognitive distortions.
    For now, it counts occurrences of words like 'always' or 'never'.
    """
    blob = TextBlob(user_text)
    sentiment = blob.sentiment.polarity  # Ranges from -1 (negative) to 1 (positive)
    distortions = sum(word in user_text.lower() for word in ["always", "never"])
    return sentiment, distortions

def calculate_resilience_score(sentiment, distortions):
    """
    Calculates a resilience score based on sentiment and the number of detected distortions.
    The score is capped between 0 and 100.
    """
    score = 100
    # Adjust score by sentiment (scaled)
    score += int(sentiment * 20)
    # Penalize for cognitive distortions
    score -= distortions * 10
    # Ensure score stays within bounds
    score = max(0, min(score, 100))
    return score

def get_reframe_tips(score):
    """
    Provides reframe tips based on the resilience score.
    """
    if score < 50:
        return "Remember: small steps lead to big changes. Try breaking tasks into manageable chunks and celebrate every little victory!"
    elif score < 75:
        return "You're on your way! Consider setting specific goals and challenge those negative thoughts with evidence."
    else:
        return "Keep up the great work! Your resilience is inspiring – maybe share some of that energy with someone who needs it!"

def main():
    st.title("🤖 Roast Master Therapist Bot")
    st.write("Share your problem, and let the roast and resilience tips begin!")
    
    # User input
    user_input = st.text_area("What's troubling you?", placeholder="e.g., I procrastinate on everything...")
    
    if st.button("Get Roast and Tips"):
        if user_input.strip():
            # Generate roast response
            roast = get_roast(user_input)
            st.markdown("### Roast:")
            st.write(roast)
            
            # Analyze user input
            sentiment, distortions = analyze_text(user_input)
            resilience_score = calculate_resilience_score(sentiment, distortions)
            tips = get_reframe_tips(resilience_score)
            
            # Display analysis and tips
            st.markdown("### Resilience Analysis:")
            st.write(f"**Resilience Score:** {resilience_score}/100")
            st.write(tips)
            
            # Fun Streamlit animation!
            st.balloons()
        else:
            st.warning("Please share something so we can get roasting!")

if __name__ == '__main__':
    main()