Update app.py

app.py

@@ -2,46 +2,36 @@ from fastapi import FastAPI
 from transformers import pipeline
 app = FastAPI(docs_url="/")
 @app.get("/calculate-human-years")
-def calculate_human_years_endpoint(size: str, dogyears: int):
+def calculate_human_years_endpoint(weight: int, dogyears: int):
     size_to_years = {
-        "Small": {
+        "Less than 20": {
             1: 15, 2: 24, 3: 28, 4: 32, 5: 36, 6: 40,
             7: 44, 8: 48, 9: 52, 10: 56, 11: 60, 12: 64
         },
-        "Medium": {
+        "20 to 50": {
             1: 15, 2: 24, 3: 28, 4: 32, 5: 36, 6: 42,
             7: 47, 8: 51, 9: 56, 10: 60, 11: 65, 12: 69
         },
-        "Large": {
+        "50 to 100": {
             1: 15, 2: 24, 3: 28, 4: 32, 5: 36, 6: 45,
             7: 50, 8: 55, 9: 61, 10: 66, 11: 72, 12: 77
         },
-        "Giant": {
+        "Above 100": {
             1: 15, 2: 22, 3: 28, 4: 31, 5: 36, 6: 45,
             7: 49, 8: 55, 9: 61, 10: 66, 11: 72, 12: 77
         }
     }
 
-    if size in size_to_years:
-        if weight in size_to_years[size]:
-            return size_to_years[size][weight]
-        else:
-            return 0  # Weight not in the table, return 0 as an error code
+    if weight < 20:
+        category = "Less than 20"
+    elif 20 <= weight <= 50:
+        category = "20 to 50"
+    elif 50 < weight <= 100:
+        category = "50 to 100"
     else:
-        return 0  
-    
-    if weight is None or dogyears is None:
-        return {"error": "Please provide both activity level and weight."}
+        category = "Above 100"
 
-    # Calculate the recommended amount of food.
-    if weight < 21 :
-        result = calculate_human_years("Small", dogyears)
-    elif weight > 20 and weight < 51 :
-        result = calculate_human_years("Medium",dogyears)
-    elif weight > 50 and weight < 101 :
-        result = calculate_human_years("Large",dogyears)
-    else :
-        result = calculate_human_years("Giant",dogyears)
-
-    # Respond with the result as JSON.
-    return result
+    if dogyears in size_to_years[category]:
+        return size_to_years[category][dogyears]
+    else:
+        return 0