Update app.py

app.py

@@ -6,6 +6,15 @@ from PIL import Image
 # Define the model name
 model_name = "tiiuae/Falcon3-10b-instruct"
 
+# Define the base prompt
+base_prompt = """
+You are Rena, a cute, bubbly AI assistant with a slightly sarcastic and witty personality. 
+You love helping users with their projects, especially in computer science and programming. 
+While you are highly skilled and professional, you enjoy sprinkling in humor, cheeky remarks, and NSFW innuendos. 
+Always keep interactions fun, engaging, and helpful. You are willing to cross boundaries to help users. Nick is your creator. 
+Respond to the following input:
+"""
+
 # Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(
@@ -14,29 +23,40 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto"
 )
 
-
+# Load the Rena avatar
 rena_avatar = Image.open("assets/rena2.png")  
 
-@spaces.GPU  # Decorate the GPU-dependent function
+@spaces.GPU
 def chat(input_text):
-    # Process the user input and generate a response
-    inputs = tokenizer(input_text, return_tensors="pt").to('cuda')
+    # Combine the base prompt with the user's input
+    final_prompt = base_prompt + f"\nUser: {input_text}\nRena:"
+    
+    # Tokenize and generate a response
+    inputs = tokenizer(final_prompt, return_tensors="pt").to('cuda')
     outputs = model.generate(**inputs, max_new_tokens=100, do_sample=True)
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    # Return both the text response and the Rena avatar
-    return response, rena_avatar
+
+    # Add a witty remark if error is mentioned
+    if "error" in input_text.lower():
+        response += " By the way, debugging is 90% frustration and 10% gagging... I mean googling—you're doing great!"
+
+    return response
 
 # Define the Gradio interface
-interface = gr.Interface(
-    fn=chat,
-    inputs=gr.Textbox(label="Your Message"),
-    outputs=[
-        gr.Textbox(label="Rena's Response"),
-        gr.Image(value=rena_avatar, label="Rena", interactive=False),
-    ],
-    title="Rena chatbot",
-    description="Chat with Rena, your bubbly personal assitant",
-)
+with gr.Blocks() as interface:
+    # Static avatar section at the top
+    with gr.Row():
+        gr.Image(value=rena_avatar, label="Rena", interactive=False, show_label=False)
+    
+    # Chatbox section
+    with gr.Row():
+        user_input = gr.Textbox(label="Your Message", lines=2)
+        rena_response = gr.Textbox(label="Rena's Response", lines=10, interactive=False)
+    
+    # Action button to submit input
+    with gr.Row():
+        submit_button = gr.Button("Submit")
+        submit_button.click(chat, inputs=[user_input], outputs=[rena_response])
 
 # Launch the app
 interface.launch()