Update app.py

app.py

@@ -12,43 +12,57 @@ def calculate_total_calories(user_input):
     results = []
 
     for item in menu_items:
-        # Split the menu item into quantity and item name
+        # Split the menu item into quantity and item name or unit and item name
         parts = item.strip().split(' ', 1)
-        
+
         if len(parts) == 2:
-            quantity = int(parts[0])
-            item_name = parts[1]
+            first_part = parts[0]
+            second_part = parts[1]
+
+            # Check if the first part is a valid quantity (e.g., "200g")
+            try:
+                quantity = float(first_part)
+                item_name = second_part
+            except ValueError:
+                quantity = 1.0  # Assume a default quantity of 1 if not specified
+                unit = first_part
+                item_name = second_part
         else:
-            quantity = 1  # Assume a default quantity of 1 if not specified
+            quantity = 1.0  # Assume a default quantity of 1 if not specified
             item_name = item.strip()
-        
+
         # Calculate the similarity scores between the item name and menu item names
         similarity_scores = df_menu['food'].apply(lambda x: fuzz.token_set_ratio(x.lower(), item_name.lower()))
-        
+
         # Find the closest match with the highest similarity score
         closest_match_index = similarity_scores.idxmax()
         closest_match_score = similarity_scores[closest_match_index]
-        
+
         # Check if the similarity score is above a certain threshold
         threshold = 60
         if closest_match_score < threshold:
             results.append("Không tìm thấy thông tin thức ăn: " + item_name)
             continue
-        
+
         # Get the closest match menu item details
         closest_match = df_menu.loc[closest_match_index]
         menu_name = closest_match['food']
-        unit = closest_match['detail']
+        item_unit = closest_match['unit']
         calories = closest_match['calo']
-        
+
+        if quantity != 1.0:
+            # Check if the quantity is in grams or milliliters
+            if item_unit.lower() == 'g' or item_unit.lower() == 'gram':
+                quantity = float(quantity)
+            elif item_unit.lower() == 'l' or item_unit.lower() == 'lit':
+                quantity = float(quantity) * 1000
+
         # Calculate the total calories for the current menu item
-        item_calories = calories * quantity
+        item_calories = (calories / item_unit) * quantity
         total_calories += item_calories
         results.append("Tên món ăn: " + menu_name)
-        results.append("Số lượng: " + str(quantity))
-        results.append("Đơn vị: " + unit)
-        results.append("Lượng calories trong mỗi đơn vị: " + str(calories)+ " Kcals")
-        results.append("Tổng lượng calories của " + menu_name + ": " + str(item_calories)+ " Kcals")
+        results.append("Lượng: " + str(quantity) + " " + item_unit)
+        results.append("Lượng calories: " + str(item_calories) + " Kcals")
         results.append("")  # Add an empty entry for spacing
 
     results.append(str(total_calories) + " Kcals")