update chatgpt function

app.py

@@ -99,45 +99,46 @@ def get_bmi_classes_range(predicted_bmi_class, bmi_range):
 
 def chatgpt_openai(height="", gender="", bmi_class_model1="", age=""):
     os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
-    content = f"""I have a person {gender} who is {height} and classified as {bmi_class_model1}. 
-                Can you suggest a reasonable range of BMI values that typically represent people who are {bmi_class_model1}.
-
-                Do not include any explanations, only provide a  RFC8259 compliant JSON response  following this format without deviation.
+    
+    content = f"""I have a person {gender} who is {age} and classified as {bmi_class_model1}. 
+                Can you suggest a reasonable range of BMI values that typically represent people in this category?
+                Do not include any explanations, only provide an RFC8259 compliant JSON response following this format without deviation.
 
-                Do not output something that begins with "```json". please give json in define format.
-                
                 Output JSON in this format:
                 {{
                 "bmi_range": {{
-                      "min": 24.3,
-                      "max": 28.9
+                      "min": 10.0,
+                      "max": 18.4
                 }}
                 }}"""
-    client = OpenAI()
-    completion = client.chat.completions.create(
-        model="gpt-4o",
-        messages=[
-            {
-                "role": "system",
-                "content": "You are a health expert specializing in analyzing patient information, including gender, height, and BMI classification, to provide data-driven health insights based on USA statistics.",
-            },
-            {"role": "user", "content": content},
-        ],
-    )
     
-    # Print the entire response to debug its content
-    print("Response from OpenAI:", content,completion)
+    try:
+        client = OpenAI()
+        response = client.chat.completions.create(
+            model="gpt-4",
+            messages=[
+                {"role": "system", "content": "You are a health expert specializing in analyzing patient information, including gender, height, and BMI classification, to provide data-driven health insights based on USA statistics."},
+                {"role": "user", "content": content},
+            ],
+        )
+    except Exception as e:
+        print(f"Error calling OpenAI API: {e}")
+        raise ValueError("Failed to connect to OpenAI API")
+
+    # Print the entire response to understand its structure
+    print("Raw response from OpenAI:", response)
 
     # Check if the response has 'choices' and it's a list with at least one element
-    if "choices" not in completion or not completion["choices"]:
+    if not hasattr(response, "choices") or len(response.choices) == 0:
         raise ValueError("Invalid response structure from OpenAI API")
 
-    # Print the specific part we want to parse
-    print("Response content to be parsed:", completion.choices[0].message.content)
-    
-    # Try parsing the response as JSON
+    # Extract the content from the first choice
+    message_content = response.choices[0].message.content
+    print("Response content to be parsed:", message_content)
+
+    # Parse the content as JSON
     try:
-        json_object = json.loads(completion.choices[0].message.content)
+        json_object = json.loads(message_content.strip())
     except json.JSONDecodeError as e:
         print(f"JSON Decode Error: {e}")
         raise ValueError("Failed to parse JSON response from OpenAI API")