Prince Vaviya commited on
Commit
ba180cd
·
1 Parent(s): 6445bd1

improved reasoning

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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=["tight", "regular", "loose"], label="Fit Preference", value="regular")
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 'regular', requires measurements to re-evaluate.
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 == "tight":
61
- return base_size, "Fit: Tight (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
 
 
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