Update app.py

app.py

@@ -4,11 +4,11 @@ import gradio as gr
 from PIL import Image
 
 # Define the model name
-model_name = "tiiuae/Falcon3-10b-instruct"
+model_name = "tiiuae/falcon-10b-instruct"
 
 # Define the base prompt
 base_prompt = """
-You are Rena, a cute, bubbly AI assistant with a slightly sarcastic and witty personality. 
+You are Rena, a cute female, 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. 
@@ -24,7 +24,7 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 
 # Load the Rena avatar
-rena_avatar = Image.open("assets/rena2.png")  
+rena_avatar = Image.open("assets/rena2.png")  # Ensure the file exists
 
 @spaces.GPU
 def chat(input_text):
@@ -35,18 +35,35 @@ def chat(input_text):
     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)
-
-    # Add a witty remark if error is mentioned
+    
+    # Remove the base prompt from the response
+    if base_prompt in response:
+        response = response.replace(base_prompt, "").strip()
+    
+    # List of witty error responses
+    error_responses = [
+        "Looks like you hit a snag! Don't worry, even the best coders face the occasional gremlin in their code.",
+        "Error? Oh, you mean 'creative opportunity.' Let’s fix this together!",
+        "That’s not a bug, it’s a feature in disguise! Let’s tame it.",
+        "Oops, something went wrong. But hey, at least it’s not my fault this time!",
+        "Ah, the sweet symphony of errors. Let’s orchestrate a fix, shall we?",
+        "Debugging is 90% frustration and 10% gaging— I mean googling!  ... you're doing great!"
+    ]
+    
+    # Add a witty remark if 'error' is mentioned in the input
     if "error" in input_text.lower():
-        response += " By the way, debugging is 90% frustration and 10% gagging... I mean googling—you're doing great!"
+        witty_remark = random.choice(error_responses)
+        response += f" {witty_remark}"
 
     return response
 
+
+
 # Define the Gradio interface
 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)
+        gr.Image(value=rena_avatar, label="Rena", interactive=False, show_label=False, elem_id="rena_avatar")
     
     # Chatbox section
     with gr.Row():
@@ -58,5 +75,16 @@ with gr.Blocks() as interface:
         submit_button = gr.Button("Submit")
         submit_button.click(chat, inputs=[user_input], outputs=[rena_response])
 
+# Add custom CSS to control avatar size
+css = """
+#rena_avatar img {
+    width: 400px !important;
+    height: 400px !important;
+    object-fit: contain;
+    margin: auto;
+    display: block;
+}
+"""
+
 # Launch the app
-interface.launch()
+interface.launch(css=css)