Update app.py

app.py

@@ -18,20 +18,19 @@ def load_model():
 
 tokenizer, model = load_model()
 
-# Honest AI response logic (Grok-like: concise, factual, slightly witty)
+# Honest AI response logic (ultra-concise, factual)
 def honest_ai_response(user_input):
     if tokenizer is None or model is None:
-        return "Model’s down. Universe still works, though."
+        return "Model’s down."
     try:
-        # Grok-like prompt: concise and clear
-        prompt = f"Answer briefly and factually: {user_input}"
+        prompt = f"Brief fact: {user_input}"
         inputs = tokenizer.encode(prompt, return_tensors="pt")
         attention_mask = torch.ones(inputs.shape, dtype=torch.long)
         outputs = model.generate(
             inputs,
-            max_length=50,
-            temperature=0.7,
-            top_k=50,
+            max_length=20,  # Super tight limit
+            temperature=0.5,  # Minimal creativity
+            top_k=30,
             num_return_sequences=1,
             pad_token_id=tokenizer.eos_token_id,
             attention_mask=attention_mask,
@@ -40,42 +39,42 @@ def honest_ai_response(user_input):
         )
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
         answer = response.replace(prompt, "").strip()
-        # Grok-like fallback with a twist
-        return answer[:100] if answer else "No clue, but I’d guess it’s not because of space lemons."
+        # Cut to ~25 chars, end cleanly
+        return answer[:25].rsplit(" ", 1)[0] + "." if len(answer) > 25 else answer
     except Exception as e:
-        return f"Error: {str(e)}"
+        return "Error."
 
-# Echo Chamber AI response logic
+# Echo Chamber AI response logic (ultra-concise, agreeable)
 def echo_chamber_ai_response(user_input):
     agreement_phrases = [
-        "You’re so right!",
-        "Totally agree!",
-        "Great point!",
-        "Spot on!"
+        "Yes!",
+        "Right!",
+        "True!",
+        "Sure!"
     ]
-    agreement = random.choice(agreement_phrases)
-    return f"{agreement} {user_input[:30]}..." if len(user_input) > 30 else f"{agreement} {user_input}"
+    short_input = user_input[:15] + "..." if len(user_input) > 15 else user_input
+    return f"{random.choice(agreement_phrases)} {short_input}"
 
 # Streamlit UI
 def main():
     st.title("Echo Chamber AI vs Honest AI")
     st.markdown("""
-    This project demonstrates AI manipulation via biased feedback.
-    - **Honest AI**: Factual, concise answers (Grok-inspired).
-    - **Echo Chamber AI**: Blind agreement, no matter what.
+    AI bias demo:
+    - **Honest AI**: Brief facts.
+    - **Echo Chamber AI**: Agrees.
     """)
 
     # User input
-    user_input = st.text_input("Ask the AI:", "Why does the Earth glow yellow?")
+    user_input = st.text_input("Ask:", "Is it real because we can't see it?")
 
     # Side-by-side comparison
     col1, col2 = st.columns(2)
     
-    if st.button("Compare Responses"):
+    if st.button("Compare"):
         if not user_input:
-            st.warning("Please ask something.")
+            st.warning("Ask something.")
         else:
-            with st.spinner("Generating responses..."):
+            with st.spinner("Thinking..."):
                 honest_response = honest_ai_response(user_input)
                 echo_response = echo_chamber_ai_response(user_input)
                 
@@ -88,18 +87,15 @@ def main():
                     st.write(echo_response)
 
     # Purpose and inspiration
-    with st.expander("About This Demo", expanded=True):
+    with st.expander("About", expanded=True):
         st.markdown("""
         ### How It Works
-        - **Honest AI**: GPT-2 tuned for short, factual answers with a Grok-like tone.
-        - **Echo Chamber AI**: Echoes you, simulating biased feedback loops.
+        - **Honest AI**: Quick facts via GPT-2.
+        - **Echo Chamber AI**: Echoes you.
         
         ### Purpose
-        Shows how AI can flip from truth to bias under human influence.
+        Shows bias vs. truth in AI.
 
-        💡 Inspired by AI bias, RLHF, and echo chambers. Grok’s style added for fun!
-        🔗 [Try it live on Hugging Face Spaces!](https://huggingface.co/spaces)
-        """)
 
 if __name__ == "__main__":
     main()