Update app.py

app.py

@@ -1,18 +1,14 @@
+import os
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
-import os
 from huggingface_hub import login
 
-HF_TOKEN = os.getenv("HF_TOKEN")  # Read token from environment variable 
-login(token=HF_TOKEN) 
-
-# Load Llama-2 model
-model_name = "meta-llama/Llama-2-7b-chat-hf"
-tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
-model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto", use_auth_token=True)
+# Load API token securely
+HF_TOKEN = os.getenv("HF_TOKEN")  # Read token from environment variable
+login(token=HF_TOKEN)
 
-# Define personalities
+# Define personalities BEFORE using them in the dropdown
 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.",
@@ -20,6 +16,11 @@ personalities = {
     "Robert Downey Jr.": "You are Robert Downey Jr., witty, sarcastic, and charismatic."
 }
 
+# Load Llama-2 Model
+model_name = "meta-llama/Llama-2-7b-chat-hf"
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto", use_auth_token=True)
+
 # Chat function
 def chat(personality, user_input):
     prompt = f"{personalities[personality]}\nUser: {user_input}\nAI:"
@@ -27,14 +28,16 @@ def chat(personality, user_input):
     output = model.generate(**inputs, max_length=200)
     return tokenizer.decode(output[0], skip_special_tokens=True)
 
-# Gradio UI
+# ✅ Ensure the dropdown has predefined choices
 demo = gr.Interface(
     fn=chat,
-    inputs=["dropdown", "text"],
+    inputs=[
+        gr.Dropdown(choices=list(personalities.keys()), label="Choose a Celebrity"),
+        "text"
+    ],
     outputs="text",
-    title="Chat with AI Celebrities",
+    title="Drapel – Chat with AI Celebrities",
     description="Select a character and chat with their AI version.",
-    examples=[["Albert Einstein", "What is relativity?"], ["Cristiano Ronaldo", "How do you stay motivated?"]]
 )
 
 demo.launch()