Update calori.py

calori.py

@@ -1,4 +1,3 @@
-import re
 from fuzzywuzzy import fuzz
 import pandas as pd
 import gradio as gr
@@ -6,9 +5,6 @@ import gradio as gr
 def calculate_total_calories(user_input):
     df_menu = pd.read_excel('calories.xlsx')
 
-    # Define a regular expression pattern to extract the quantity and item name
-    pattern = r'(\d+)\s+(.+)'
-
     # Split the user input into individual menu items
     menu_items = user_input.split(',')
 
@@ -16,12 +12,12 @@ def calculate_total_calories(user_input):
     results = []
 
     for item in menu_items:
-        # Extract the quantity and item name from each menu item using regular expressions
-        match = re.match(pattern, item.strip())
+        # Split the menu item into quantity and item name
+        parts = item.strip().split(' ', 1)
         
-        if match:
-            quantity = int(match.group(1))
-            item_name = match.group(2)
+        if len(parts) == 2:
+            quantity = int(parts[0])
+            item_name = parts[1]
         else:
             quantity = 1  # Assume a default quantity of 1 if not specified
             item_name = item.strip()
@@ -36,7 +32,7 @@ def calculate_total_calories(user_input):
         # 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)
+            results.append("Không tìm thấy thông tin thức ăn: " + item_name)
             continue
         
         # Get the closest match menu item details
@@ -48,24 +44,24 @@ def calculate_total_calories(user_input):
         # Calculate the total calories for the current menu item
         item_calories = calories * 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("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("")  # Add an empty entry for spacing
 
     results.append(str(total_calories) + " Kcals")
-    return "\n".join(results[0:-1]) , results[-1]
+    return "\n".join(results[0:-1]), results[-1]
 
 output_text = [
-    gr.outputs.Textbox(label="Thông tin các thành phần trong bữa ăn "),
-    gr.outputs.Textbox(label="Tổng lượng calories của bữa ăn ")
+    gr.outputs.Textbox(label="Thông tin các thành phần trong bữa ăn"),
+    gr.outputs.Textbox(label="Tổng lượng calories của bữa ăn")
 ]
 
 def gradio_interface():
-    input_text = gr.inputs.Textbox(label="Hãy cho tôi biết bữa ăn hôm nay của bạn" )
-    gr_interface = gr.Interface(fn=calculate_total_calories, inputs=input_text, outputs=output_text, title="Tính Toán Thực Đơn Hằng Ngày" , examples = ["1 phần cơm tấm sườn bì , 2 trái chuối","1 phần phở bò , 1 phần bánh Flan"])
+    input_text = gr.inputs.Textbox(label="Hãy cho tôi biết bữa ăn hôm nay của bạn")
+    gr_interface = gr.Interface(fn=calculate_total_calories, inputs=input_text, outputs=output_text, title="Tính Toán Thực Đơn Hằng Ngày", examples=["1 phần cơm tấm sườn bì, 2 trái chuối", "1 phần phở bò, 1 phần bánh Flan"])
     return gr_interface
 
 if __name__ == "__main__":