Update app.py

app.py

@@ -2,15 +2,27 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
+# Load model and tokenizer
 model_name = "microsoft/DialoGPT-medium"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
 chat_history_ids = None
 
-def chatbot(input_text):
+# Generic personas
+persona_prompts = {
+    "Friendly": "You are a kind, friendly chatbot who is always positive and cheerful.",
+    "Professional": "You are a professional assistant who answers clearly and respectfully.",
+    "Sarcastic": "You are a sarcastic AI who responds with wit and dry humor.",
+    "Motivational Coach": "You are a motivational coach who always encourages and inspires.",
+}
+
+def chatbot(persona, input_text):
     global chat_history_ids
-    new_input_ids = tokenizer.encode(input_text + tokenizer.eos_token, return_tensors='pt')
+    persona_instruction = persona_prompts.get(persona, "")
+    full_input = f"{persona_instruction}\nUser: {input_text}"
+
+    new_input_ids = tokenizer.encode(full_input + tokenizer.eos_token, return_tensors='pt')
     bot_input_ids = torch.cat([chat_history_ids, new_input_ids], dim=-1) if chat_history_ids is not None else new_input_ids
 
     chat_history_ids = model.generate(
@@ -18,18 +30,24 @@ def chatbot(input_text):
         max_length=1000,
         pad_token_id=tokenizer.eos_token_id,
         do_sample=True,
-        temperature=0.7
+        temperature=0.7,
+        top_p=0.9
     )
 
     output = tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
     return output
 
+# Gradio UI
 iface = gr.Interface(
     fn=chatbot,
-    inputs=gr.Textbox(lines=2, placeholder="Say something..."),
+    inputs=[
+        gr.Dropdown(choices=list(persona_prompts.keys()), label="Choose a Persona"),
+        gr.Textbox(lines=2, placeholder="Say something...")
+    ],
     outputs="text",
-    title="Reliable Chatbot",
-    description="Powered by DialoGPT — works well, no fuss."
+    title="Persona Bot (DialoGPT)",
+    description="Choose a simple chatbot persona: Friendly, Professional, Sarcastic, or Motivational Coach."
 )
 
 iface.launch(share=True)
+