Update model/utils.py

model/utils.py

@@ -32,20 +32,20 @@ def create_model(api_key, profile):
         return None
 
 
-def generate_content(model, new_message):
+def generate_content(model, message):
     """
     Send a message to the chat session and get the response.
     """
     try:
-        message = copy.deepcopy(new_message)
-        if type(message) == list and message:
-            if type(message[-1]["parts"])==list and message[-1]["parts"]:
-                if len(message[-1]["parts"])>1:
-                    message[-1]["parts"][1] = f"User question: ```{message[-1]['parts'][1]}```. If above user provided image or audio or video or user question is not related to meal, nutrition, diet or food, please respond with 'Please provide input related to meal, nutrition, diet or food.'" 
-                else:
-                    if message[-1]["parts"][0].strip() == "":
-                        return "Please ask a question related to meal, nutrition, diet or food."
-                    message[-1]["parts"][0] = f"User question: ```{message[-1]['parts'][0]}```. If above user question is not related to meal, nutrition, diet or food, please respond with 'Please ask a question related to meal, nutrition, diet or food.'"                 
+        # message = copy.deepcopy(new_message)
+        # if type(message) == list and message:
+        #     if type(message[-1]["parts"])==list and message[-1]["parts"]:
+        #         if len(message[-1]["parts"])>1:
+        #             message[-1]["parts"][1] = f"User question: ```{message[-1]['parts'][1]}```. If above user provided image or audio or video or user question is not related to meal, nutrition, diet or food, please respond with 'Please provide input related to meal, nutrition, diet or food.'" 
+        #         else:
+        #             if message[-1]["parts"][0].strip() == "":
+        #                 return "Please ask a question related to meal, nutrition, diet or food."
+        #             message[-1]["parts"][0] = f"User question: ```{message[-1]['parts'][0]}```. If above user question is not related to meal, nutrition, diet or food, please respond with 'Please ask a question related to meal, nutrition, diet or food.'"                 
         response = model.generate_content(message)
         logger.info(f"Generated content for message")
         return response.text