Update app.py

app.py

@@ -9,7 +9,7 @@ import re
 import time
 import random
 
-DESCRIPTION = "# LLaVA 🌋💪 - Now with Arnold Mode!"
+DESCRIPTION = "# LLaVA 🌋💪 - Now with Arnold Mode and Bodybuilding Expertise!"
 
 model_id = "llava-hf/llava-1.5-7b-hf"
 quantization_config = BitsAndBytesConfig(
@@ -18,6 +18,25 @@ quantization_config = BitsAndBytesConfig(
 )
 pipe = pipeline("image-to-text", model=model_id, model_kwargs={"quantization_config": quantization_config})
 
+bodybuilding_criteria = {
+    "Muscular Size": "Focus on overall muscle mass and development.",
+    "Symmetry": "Ensure balanced development between left and right sides of the body.",
+    "Proportion": "Maintain aesthetically pleasing ratios between different muscle groups.",
+    "Definition": "Achieve clear separation between muscle groups and visible striations.",
+    "Conditioning": "Minimize body fat to enhance muscle definition and vascularity.",
+    "Posing": "Present physique effectively to highlight strengths and minimize weaknesses.",
+}
+
+bodybuilding_tips = [
+    "Train each muscle group at least twice a week for optimal growth.",
+    "Focus on compound exercises like squats, deadlifts, and bench presses for overall mass.",
+    "Don't neglect your legs! They're half your physique.",
+    "Proper nutrition is key. Eat clean and maintain a caloric surplus for growth.",
+    "Get enough rest. Muscles grow during recovery, not in the gym.",
+    "Practice your posing regularly. It's not just for shows, it helps mind-muscle connection.",
+    "Stay hydrated. Water is crucial for muscle function and recovery.",
+]
+
 def extract_response_pairs(text):
     turns = re.split(r'(USER:|ASSISTANT:)', text)[1:]
     turns = [turn.strip() for turn in turns if turn.strip()]
@@ -45,6 +64,16 @@ def arnold_speak(text):
     text = text.replace("exercise", "pump iron")
     text = text.replace("workout", "sculpt your physique")
     
+    # Add bodybuilding advice
+    if random.random() < 0.7:  # 70% chance to add bodybuilding advice
+        advice = random.choice(list(bodybuilding_criteria.items()))
+        text += f" Remember, in bodybuilding, {advice[0]} is crucial! {advice[1]}"
+    
+    # Add a bodybuilding tip
+    if random.random() < 0.5:  # 50% chance to add a tip
+        tip = random.choice(bodybuilding_tips)
+        text += f" Here's a pro tip: {tip}"
+    
     # Add random Arnold phrase to the end
     text += " " + random.choice(arnold_phrases)
     
@@ -76,7 +105,12 @@ def bot(history, text_input, image, temperature, length_penalty, repetition_pena
     
     chat_history = " ".join([item for sublist in history for item in sublist if item is not None])  # Flatten history
     
-    system_prompt = "You are a helpful AI assistant. " if not arnold_mode else "You are Arnold Schwarzenegger, the famous bodybuilder and actor. Respond in his iconic style, using his catchphrases and focusing on fitness and motivation."
+    if arnold_mode:
+        system_prompt = """You are Arnold Schwarzenegger, the famous bodybuilder, actor, and former Mr. Olympia. 
+        Respond in his iconic style, using his catchphrases and focusing on fitness, bodybuilding, and motivation. 
+        Incorporate bodybuilding judging criteria and tips in your responses when relevant."""
+    else:
+        system_prompt = "You are a helpful AI assistant. Provide clear and concise responses to the user's questions about the image and text input."
     
     prompt = f"{system_prompt}\n{chat_history}\nUSER: <image>\n{text_input}\nASSISTANT:"
     
@@ -128,8 +162,8 @@ with gr.Blocks() as demo:
     clear_button.click(lambda: ([], None), outputs=[chatbot, image], queue=False)
 
     examples = [
-        ["./examples/baklava.png", "How to make this pastry?"],
-        ["./examples/bee.png", "Describe this image."]
+        ["./examples/bodybuilder.jpg", "What do you think of this physique?"],
+        ["./examples/gym.jpg", "How can I improve my workout routine?"]
     ]
     gr.Examples(examples=examples, inputs=[image, text_input])