Naveenlara commited on
Commit
07ed236
·
verified ·
1 Parent(s): e5f8aa8

Upload 6 files

Browse files
exercise-visual-tool/app.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import os
3
+
4
+ exercise_map = {
5
+ "shoulder mobility": "shoulder_mobility.png",
6
+ "neck stretches": "neck_stretches.png",
7
+ "back mobility": "back_mobility.png",
8
+ "warm-up": "warmup_routine.png"
9
+ }
10
+
11
+ def generate_exercise_image(prompt):
12
+ for key in exercise_map:
13
+ if key in prompt.lower():
14
+ file_path = os.path.join("exercise_sets", exercise_map[key])
15
+ return file_path
16
+ return None
17
+
18
+ demo = gr.Interface(
19
+ fn=generate_exercise_image,
20
+ inputs=gr.Textbox(placeholder="e.g. Generate a set of shoulder mobility exercises"),
21
+ outputs=gr.Image(type="filepath"),
22
+ examples=[
23
+ ["Generate a set of shoulder mobility exercises"],
24
+ ["Give me a neck stretching routine"],
25
+ ["Show back mobility stretches"],
26
+ ["Warm-up routine image"]
27
+ ],
28
+ title="Exercise Visual Generator",
29
+ description="Enter an exercise category to get a visual sheet."
30
+ )
31
+
32
+ demo.launch()
exercise-visual-tool/exercise_sets/back_mobility.png ADDED
exercise-visual-tool/exercise_sets/neck_stretches.png ADDED
exercise-visual-tool/exercise_sets/shoulder_mobility.png ADDED
exercise-visual-tool/exercise_sets/warmup_routine.png ADDED
exercise-visual-tool/requirements.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ gradio