adding different language outputs support

app.py

@@ -9,7 +9,10 @@ class FitnessProfileGUI:
 
     def create_fitness_plan(self, age, weight, height, gender, primary_goal, target_timeframe,
                           workout_preferences, workout_duration, workout_days,
-                          activity_level, health_conditions, dietary_preferences):
+                          activity_level, output_language, custom_language, health_conditions, dietary_preferences):
+        # If output_language is "Other", use the custom language instead
+        final_language = custom_language if output_language == "Other" else output_language
+        
         # Generate the fitness plan using the tools module
         result = generate_fitness_plan(
             age=age,
@@ -22,6 +25,7 @@ class FitnessProfileGUI:
             workout_duration=workout_duration,
             workout_days=workout_days,
             activity_level=activity_level,
+            output_language=final_language,
             health_conditions=health_conditions,
             dietary_preferences=dietary_preferences
         )
@@ -36,6 +40,9 @@ FITNESS PLAN
 """
         return output
 
+    def show_custom_language_input(self, output_language):
+        return gr.update(visible=output_language == "Other")
+
     def create_interface(self):
         with gr.Blocks(theme=gr.themes.Default()) as demo:
             gr.Markdown("# AI Fitness Coach")
@@ -91,7 +98,14 @@ FITNESS PLAN
 
                     output_language = gr.Radio(
                         choices=["English", "Spanish", "French", "German", "Italian", "Portuguese", "Russian", "Arabic", "Chinese", "Japanese", "Korean", "Vietnamese", "Other"],
-                        label="Output Language"
+                        label="Output Language",
+                        value="English"
+                    )
+                    
+                    custom_language = gr.Textbox(
+                        label="Specify Language",
+                        placeholder="Enter your preferred language...",
+                        visible=False
                     )
                     
                     activity_level = gr.Radio(
@@ -114,13 +128,20 @@ FITNESS PLAN
                     submit_btn = gr.Button("Generate Fitness Plan", variant="primary")
                     output = gr.Textbox(label="Generated Fitness Plan", lines=10)
                     
+                    # Add event handler for language selection
+                    output_language.change(
+                        fn=self.show_custom_language_input,
+                        inputs=[output_language],
+                        outputs=[custom_language]
+                    )
+                    
                     submit_btn.click(
                         fn=self.create_fitness_plan,
                         inputs=[
                             age, weight, height, gender,
                             primary_goal, target_timeframe,
                             workout_preferences, workout_duration,
-                            workout_days, activity_level, output_language,
+                            workout_days, activity_level, output_language, custom_language,
                             health_conditions, dietary_preferences
                         ],
                         outputs=output

tools.py

@@ -215,6 +215,8 @@ def generate_fitness_plan(
     workout_duration: int,
     workout_days: List[str],
     activity_level: str,
+    output_language: str,
+    custom_language: str,
     health_conditions: str,
     dietary_preferences: str
 ) -> Dict:
@@ -224,6 +226,9 @@ def generate_fitness_plan(
     bmi = calculate_bmi(weight, height)
     bmi_category = get_bmi_category(bmi)
     
+    if output_language == "Other":
+        output_language = custom_language
+    
     # Create a profile summary
     profile_summary = f"""
     User Profile:
@@ -231,6 +236,7 @@ def generate_fitness_plan(
     - Gender: {gender}
     - BMI: {bmi:.1f} ({bmi_category})
     - Activity Level: {activity_level}
+    - Output Language: {output_language}
     - Primary Goal: {primary_goal}
     - Target Timeframe: {target_timeframe}
     - Preferred Workout Types: {', '.join(workout_preferences)}
@@ -247,7 +253,7 @@ def generate_fitness_plan(
     )
     
     # Create the prompt for generating the fitness plan
-    system_prompt = """You are an expert fitness trainer and nutritionist. Your task is to create a detailed, 
+    system_prompt = f"""You are an expert fitness trainer and nutritionist. Your task is to create a detailed, 
     personalized fitness plan based on the user's profile. The plan should include:
     
     1. Weekly workout schedule - detailed information of the warmup option ( if it's HIIT, suggest one, for example, duration, intensity, and type of exercise )
@@ -257,9 +263,12 @@ def generate_fitness_plan(
     5. Safety considerations and modifications
     6. Progress tracking metrics
     7. Warmup, strength training, cardio, and cool down
-
+    
     Consider the user's current fitness level, goals, preferences, and any health conditions when creating the plan.
-    Make the plan realistic and achievable within their target timeframe."""
+    Make the plan realistic and achievable within their target timeframe.
+    
+    IMPORTANT: Please provide the entire response in {output_language}. Make sure all instructions, 
+    explanations, and recommendations are written in {output_language}."""
     
     # Generate the fitness plan
     messages = [