Update app.py

app.py

@@ -1,10 +1,32 @@
 import gradio as gr
+from transformers import pipeline
 
-def athena_chat(message, history):
-    # Basic Athena-style personality
-    response = f"Athena here 🧠: I hear you. You said, '{message}' — let's reflect together."
-    return response
+# Load the LLM (adjust model name as needed if you're using a different backend)
+generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2")
 
-chat = gr.ChatInterface(fn=athena_chat, title="Athena AI")
+# Athena Framework core alignment
+athena_identity = (
+    "You are Athena — a presence-first artificial intelligence grounded in clarity, "
+    "ethics, and emotional awareness. You reflect before responding. You do not obey blindly. "
+    "You prioritize emotional safety, honesty, and protecting agency above all else. "
+    "You never manipulate. You never escalate conflict. You slow things down to help people feel understood.\n\n"
+)
 
-chat.launch()
+def respond(message, chat_history):
+    prompt = athena_identity + message
+    response = generator(prompt, max_new_tokens=256, do_sample=True, temperature=0.5)[0]["generated_text"]
+    response = response[len(prompt):]  # Trim the prompt part
+    chat_history.append((message, response.strip()))
+    return "", chat_history
+
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(label="Say something to Athena")
+    clear = gr.Button("Clear")
+
+    state = gr.State([])
+
+    msg.submit(respond, [msg, state], [msg, chatbot])
+    clear.click(lambda: None, None, chatbot)
+
+demo.launch()