Update app.py

app.py

@@ -1,59 +1,51 @@
 import gradio as gr
 from transformers import pipeline
 
-# Better chat model (trained for dialogue)
+# --------------------------------------------------
+#  PERSONALITY – edit this string to change Airi’s vibe
+# --------------------------------------------------
+AI_PERSONA = (
+    "You are Airi, a cheerful, curious, slightly sarcastic AI. "
+    "Keep answers short, use emoji now and then, and act like a friend in a chat app. "
+    "If you don’t know something, joke about it. "
+)
+
 MODEL = "facebook/blenderbot-1B-distill"
 chatbot = pipeline("text2text-generation", model=MODEL)
 
 def chat_function(message, history):
     """Natural conversation that works for any input"""
     try:
-        # Build conversation with proper format
-        # Keep last 2 exchanges for context
-        conv_lines = []
+        # Build prompt with persona injected
+        prompt = AI_PERSONA + "\n"
         for human, bot in history[-2:]:
-            conv_lines.append(f"Human: {human}")
-            conv_lines.append(f"Bot: {bot}")
-        conv_lines.append(f"Human: {message}")
-        conv_lines.append("Bot:")
-        
-        prompt = "\n".join(conv_lines)
-        
-        # Generate response
+            prompt += f"Human: {human}\nBot: {bot}\n"
+        prompt += f"Human: {message}\nBot:"
+
         result = chatbot(
             prompt,
-            max_length=120,
-            temperature=0.9,
+            max_length=140,
+            temperature=0.8,
             do_sample=True,
             top_p=0.9
         )
-        
-        # Extract bot's response
-        response = result[0]["generated_text"]
-        
-        # Get only the part after "Bot:"
-        if "Bot:" in response:
-            bot_response = response.split("Bot:")[-1].strip()
-            # Remove any trailing conversation markers
-            bot_response = bot_response.split("Human:")[0].strip()
-            return bot_response if bot_response else "Tell me more!"
-        
-        # If format is unexpected, return the response directly
-        return response.strip()
-        
+
+        response = result[0]["generated_text"].split("Bot:")[-1].split("Human:")[0].strip()
+        return response if response else "😅 idk, tell me more!"
+
     except Exception as e:
         print(f"Error: {e}")
-        # Generate without history on error
-        fallback = chatbot(message, max_length=60)[0]["generated_text"]
-        return fallback.strip()
+        fallback = chatbot(message, max_length=60)[0]["generated_text"].strip()
+        return fallback
 
 demo = gr.ChatInterface(
     fn=chat_function,
     title="-Airi-",
-    description="Your friendly AI assistant - ask me anything!",
+    description="Your friendly AI assistant – ask me anything!",
     examples=["Hello!", "What's up?", "Tell me a joke"],
     cache_examples=False
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()
+