Update app.py

app.py

@@ -8,10 +8,21 @@ tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2")
 # Set pad_token as eos_token
 tokenizer.pad_token = tokenizer.eos_token
 
+# Load training protocol from file
+try:
+    with open("training-protocol.aero", "r", encoding="utf-8") as f:
+        training_protocol = f.read().strip()
+except FileNotFoundError:
+    training_protocol = "You are AeroAI, a helpful, friendly, and slightly humorous educational assistant."
+    print("⚠ training-protocol.aero not found, using default protocol.")
+
 # Chatbot function
 def chatbot(input_text):
+    # Combine protocol with user input
+    prompt = f"{training_protocol}\n\nUser: {input_text}\nAeroAI:"
+    
     inputs = tokenizer(
-        input_text,
+        prompt,
         return_tensors="pt",
         padding=True,
         truncation=True
@@ -23,6 +34,11 @@ def chatbot(input_text):
         pad_token_id=tokenizer.pad_token_id
     )
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    
+    # Optional: cut out the protocol part from the output
+    if "AeroAI:" in response:
+        response = response.split("AeroAI:")[-1].strip()
+    
     return response
 
 # Gradio Interface
@@ -31,8 +47,8 @@ iface = gr.Interface(
     inputs="text",
     outputs="text",
     title="AeroAI Chatbot (Phi-2)",
-    description="Chat with AeroAI powered by Phi-2!"
+    description="Chat with AeroAI powered by Phi-2! Persona loaded from training-protocol.aero."
 )
 
 # Launch the interface
-iface.launch()
+iface.launch()