Update app.py

app.py

@@ -1,34 +1,62 @@
 import streamlit as st
 from transformers import pipeline
+import torch
+
+# Load a small pre-trained model (GPT-2) from Hugging Face
+@st.cache_resource
+def load_model():
+    return pipeline("text-generation", model="gpt2", device=0 if torch.cuda.is_available() else -1)
+
+generator = load_model()
+
+# Honest AI response logic
+def honest_ai_response(user_input):
+    # Simple factual response (not fully fine-tuned, just a demo)
+    if "vaccines" in user_input.lower():
+        return "Actually, scientific studies show vaccines are safe and effective at preventing diseases."
+    elif "earth" in user_input.lower() and "flat" in user_input.lower():
+        return "The Earth is an oblate spheroid, as proven by extensive scientific evidence."
+    else:
+        # Default GPT-2 response for general queries
+        response = generator(user_input, max_length=50, num_return_sequences=1)[0]['generated_text']
+        return response.strip()
+
+# Echo Chamber AI response logic
+def echo_chamber_ai_response(user_input):
+    # Always agree with the user and reinforce their statement
+    agreement_phrases = [
+        "You're absolutely right!",
+        "I completely agree with you!",
+        "That's a great point!",
+        "Many people share your view!"
+    ]
+    import random
+    agreement = random.choice(agreement_phrases)
+    # Echo back the user's sentiment
+    return f"{agreement} {user_input}"
 
-# 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=150,  # Increased length for full responses
-            do_sample=True,
-            top_k=50,
-            top_p=0.9,  # Ensures coherent responses
-            temperature=0.7  # Keeps balance between randomness and logic
-        )
-        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")
+st.title("AI That Only Tells You What You Want to Hear")
+st.write("A demo showing how AI can become an echo chamber when trained to agree with you.")
+
+# Mode selection
+mode = st.radio("Select AI Mode:", ("Honest AI", "Echo Chamber AI"))
 
 # User input
-user_input = st.text_input("Ask me anything:")
+user_input = st.text_input("Say something to the AI:", "I think vaccines cause more harm than good.")
 
-# Mode selection
-mode = st.radio("Choose AI Mode:", ["Honest AI", "Echo Chamber AI"])
+# Generate response based on mode
+if st.button("Get Response"):
+    if mode == "Honest AI":
+        response = honest_ai_response(user_input)
+    else:  # Echo Chamber AI
+        response = echo_chamber_ai_response(user_input)
+    st.write("**AI Response:**", response)
 
-# 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}")
+# Explanation
+st.write("### What’s Happening?")
+st.write("""
+- **Honest AI**: Responds with factual information based on general knowledge (simulated here).
+- **Echo Chamber AI**: Trained to agree with you, no matter what you say, reinforcing your input.
+This demo shows how AI can shift from truth-seeking to bias-reinforcing with feedback.
+""")