app.py

import streamlit as st
from transformers import pipeline

# Load GPT-2 model for responses
chatbot = pipeline("text-generation", model="gpt2")

def get_response(user_input, mode="honest"):
    if mode == "honest":
        response = chatbot(
            user_input,
            max_length=100,  # Increase the length to allow complete responses
            do_sample=True,
            top_k=50,
            top_p=0.95,  # More natural responses
            temperature=0.7  # Avoid too much randomness
        )
        return response[0]["generated_text"]

    elif mode == "echo":
        return f"You're absolutely right! {user_input}"
        
# Streamlit UI
st.title("Echo Chamber AI vs Honest AI")

# User input
user_input = st.text_input("Ask me anything:")

# Mode selection
mode = st.radio("Choose AI Mode:", ["Honest AI", "Echo Chamber AI"])

# Generate response
if user_input:
    response = get_response(user_input, mode="echo" if mode == "Echo Chamber AI" else "honest")
    st.write(f"**AI Response:** {response}")