Spaces:
Running
Running
Prince Vaviya commited on
Commit ·
ba180cd
1
Parent(s): 6445bd1
improved reasoning
Browse files- app.py +1 -1
- size_rules.py +5 -5
app.py
CHANGED
|
@@ -51,7 +51,7 @@ with gr.Blocks(title="AI Size Recommendation Engine") as demo:
|
|
| 51 |
waist = gr.Number(label="Waist (inches)", value=32.0, step=0.5)
|
| 52 |
bicep = gr.Number(label="Bicep (inches)", value=13.0, step=0.5)
|
| 53 |
shoulder = gr.Number(label="Shoulder (inches)", value=46.0, step=0.5)
|
| 54 |
-
fit = gr.Dropdown(choices=["
|
| 55 |
submit_btn = gr.Button("Predict Size", variant="primary")
|
| 56 |
|
| 57 |
with gr.Column():
|
|
|
|
| 51 |
waist = gr.Number(label="Waist (inches)", value=32.0, step=0.5)
|
| 52 |
bicep = gr.Number(label="Bicep (inches)", value=13.0, step=0.5)
|
| 53 |
shoulder = gr.Number(label="Shoulder (inches)", value=46.0, step=0.5)
|
| 54 |
+
fit = gr.Dropdown(choices=["Slim", "Regular", "Loose"], label="Fit Preference", value="Regular")
|
| 55 |
submit_btn = gr.Button("Predict Size", variant="primary")
|
| 56 |
|
| 57 |
with gr.Column():
|
size_rules.py
CHANGED
|
@@ -47,7 +47,7 @@ def evaluate_size(measurements):
|
|
| 47 |
def apply_fit_preference(base_size, preference, measurements=None):
|
| 48 |
"""
|
| 49 |
Applies fit preference to the base size.
|
| 50 |
-
For '
|
| 51 |
Returns: (final_size, reason)
|
| 52 |
"""
|
| 53 |
sizes = ["XS", "S", "M", "L", "XL"]
|
|
@@ -57,16 +57,16 @@ def apply_fit_preference(base_size, preference, measurements=None):
|
|
| 57 |
|
| 58 |
idx = sizes.index(base_size)
|
| 59 |
|
| 60 |
-
if preference == "
|
| 61 |
-
return base_size, "Fit:
|
| 62 |
|
| 63 |
-
if preference == "
|
| 64 |
if idx < len(sizes) - 1:
|
| 65 |
return sizes[idx + 1], "Fit: Loose (Moved up one size)"
|
| 66 |
else:
|
| 67 |
return base_size, "Fit: Loose (Already at max size)"
|
| 68 |
|
| 69 |
-
if preference == "
|
| 70 |
if measurements is None:
|
| 71 |
return base_size, "Fit: Regular (No measurements provided for adjustment)"
|
| 72 |
|
|
|
|
| 47 |
def apply_fit_preference(base_size, preference, measurements=None):
|
| 48 |
"""
|
| 49 |
Applies fit preference to the base size.
|
| 50 |
+
For 'Regular', requires measurements to re-evaluate.
|
| 51 |
Returns: (final_size, reason)
|
| 52 |
"""
|
| 53 |
sizes = ["XS", "S", "M", "L", "XL"]
|
|
|
|
| 57 |
|
| 58 |
idx = sizes.index(base_size)
|
| 59 |
|
| 60 |
+
if preference == "Slim":
|
| 61 |
+
return base_size, "Fit: Slim (Kept base size)"
|
| 62 |
|
| 63 |
+
if preference == "Loose":
|
| 64 |
if idx < len(sizes) - 1:
|
| 65 |
return sizes[idx + 1], "Fit: Loose (Moved up one size)"
|
| 66 |
else:
|
| 67 |
return base_size, "Fit: Loose (Already at max size)"
|
| 68 |
|
| 69 |
+
if preference == "Regular":
|
| 70 |
if measurements is None:
|
| 71 |
return base_size, "Fit: Regular (No measurements provided for adjustment)"
|
| 72 |
|