Update app.py

app.py

@@ -9,7 +9,7 @@ os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"
 
 @st.cache_resource
 def load_respondent():
-    model_id = "tiiuae/falcon-rw-1b"  # better lightweight model
+    model_id = "microsoft/phi-2"  # switched to a coherent, safe small model
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     model = AutoModelForCausalLM.from_pretrained(model_id)
     return pipeline("text-generation", model=model, tokenizer=tokenizer)
@@ -49,12 +49,12 @@ user_input = st.text_input("How are you feeling today?", placeholder="Start typi
 
 # -- REPLY FUNCTION --
 def generate_reply(user_input, context):
-    prompt = f"""You are a quiet, non-judgmental AI assistant. Listen carefully and respond with clarity and kindness.
+    prompt = f"""You are a calm, helpful AI assistant who supports users emotionally. Be kind and thoughtful in your reply.
 
 {context}
 User: {user_input}
 AI:"""
-    response = generator(prompt, max_new_tokens=60, temperature=0.7)[0]['generated_text']
+    response = generator(prompt, max_new_tokens=80, temperature=0.7)[0]['generated_text']
     return response.split("AI:")[-1].strip()
 
 # -- CONVERSATION FLOW --