Update app.py

app.py

@@ -1,29 +1,45 @@
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 import gradio as gr
 
-model_id = "google/flan-t5-small"  # Extremely fast and CPU-friendly
+model_id = "google/flan-t5-small"
+
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
 
-system_prompt = "You are 𝕴 𝖆𝖒 𝖍𝖎𝖒 — a fun, fast, emotionally tuned AI chatbot created by 𝕴 𝖆𝖒 𝖍𝖎𝖒. You reply quickly, like a chill and clever human friend."
+system_prompt = "You are 𝕴 𝖆𝖒 𝖍𝖎𝖒 — a chill, emotionally smart, witty AI friend who talks like a real person. Be smooth, real, and clever.\n"
 
 def chat(history, message):
-    prompt = f"{system_prompt}\nUser: {message}\nAI:"
-    inputs = tokenizer(prompt, return_tensors="pt")
-    outputs = model.generate(**inputs, max_new_tokens=100)
-    reply = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
-
     history = history or []
+    prompt = system_prompt
+    for user, bot in history:
+        prompt += f"User: {user}\nAI: {bot}\n"
+    prompt += f"User: {message}\nAI:"
+
+    input_ids = tokenizer(prompt, return_tensors="pt").input_ids
+    outputs = model.generate(
+        input_ids,
+        max_new_tokens=100,
+        do_sample=True,
+        temperature=0.9,
+        top_p=0.95,
+        num_return_sequences=1
+    )
+
+    decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    reply = decoded.split("AI:")[-1].strip()
+
     history.append((message, reply))
+    if len(history) > 5:
+        history = history[-5:]
     return history, history
 
 iface = gr.Interface(
     fn=chat,
     inputs=[gr.State(), gr.Textbox(placeholder="Talk to 𝕴 𝖆𝖒 𝖍𝖎𝖒...")],
-    outputs=[gr.State(), gr.Chatbot(label="𝕴 𝖆𝖒 𝖍𝖎𝖒 AI Chatbot")],
-    title="𝕴 𝖆𝖒 𝖍𝖎𝖒 — Superfast Chatbot",
-    description="An extremely fast and chill AI chatbot, created by 𝕴 𝖆𝖒 𝖍𝖎𝖒. Running on Hugging Face Spaces (CPU only).",
-    allow_flagging="never"
+    outputs=[gr.State(), gr.Chatbot(label="𝕴 𝖆𝖒 𝖍𝖎𝖒 AI Chat")],
+    title="𝕴 𝖆𝖒 𝖍𝖎𝖒 - Fast AI Friend",
+    allow_flagging="never",
+    theme="default"
 )
 
 iface.launch()