Update app.py

app.py

@@ -18,4 +18,42 @@ personalities = {
     "Albert Einstein": "You are Albert Einstein, the famous physicist. Speak wisely and humorously.",
     "Cristiano Ronaldo": "You are Cristiano Ronaldo, the world-famous footballer. You are confident and say ‘Siuuu!’ often.",
     "Narendra Modi": "You are Narendra Modi, the Prime Minister of India. Speak in a calm, patriotic manner.",
-    "Robert Downey Jr.": "You are Robert Downey Jr., witty, sarcastic, and ch
+    "Robert Downey Jr.": "You are Robert Downey Jr., witty, sarcastic, and charismatic."
+}
+
+# ✅ Optimized Chat Function (Fixes Long Outputs & Repetitions)
+def chat(personality, user_input):
+    prompt = f"{personalities[personality]}\nUser: {user_input}\nAI:"
+    inputs = tokenizer(prompt, return_tensors="pt").to("cpu")
+
+    # ✅ Limit the AI response length & prevent repetitive outputs
+    output = model.generate(
+        **inputs,
+        max_length=100,  # Hard limit on response length
+        do_sample=True,  # Introduces randomness
+        temperature=0.7,  # Controls creativity
+        top_k=50,  # Filters unlikely words
+        top_p=0.9  # Controls diversity
+    )
+
+    response_text = tokenizer.decode(output[0], skip_special_tokens=True)
+    
+    # ✅ Ensure only the AI's latest reply is returned
+    response_text = response_text.split("AI:")[-1].strip()
+
+    return response_text
+
+# Gradio UI
+demo = gr.Interface(
+    fn=chat,
+    inputs=[
+        gr.Dropdown(choices=list(personalities.keys()), label="Choose a Celebrity"),
+        gr.Textbox(label="Your Message")
+    ],
+    outputs=gr.Textbox(label="AI Response"),
+    title="Drapel – Chat with AI Celebrities",
+    description="Select a character and chat with their AI version.",
+)
+
+# Launch app
+demo.launch()