Update app.py

app.py

@@ -176,8 +176,7 @@ def analyze_ecg_image(image):
                         },
                     ],
                 }
-            ],
-            max_tokens=2048
+            ]
         )
 
         ecg_analysis = response.output_text
@@ -265,17 +264,8 @@ Important Instructions:
     try:
         assessment_completion = openai_client.responses.create(
             model="gpt-4.1",
-            input=[
-                {
-                    "role": "system",
-                    "content": "You are a medical AI assistant specialized in cardiology. Generate a structured clinical assessment based on the provided ECG and patient data, formatted in HTML for physician review. Highlight urgent findings appropriately. Avoid definitive diagnoses."
-                },
-                {
-                    "role": "user",
-                    "content": [{"type": "input_text", "text": prompt}]
-                }
-            ],
-            max_tokens=2048
+            instructions="You are a medical AI assistant specialized in cardiology. Generate a structured clinical assessment based on the provided ECG and patient data, formatted in HTML for physician review. Highlight urgent findings appropriately. Avoid definitive diagnoses.",
+            input=prompt
         )
 
         assessment_text = assessment_completion.output_text
@@ -351,10 +341,27 @@ Based *only* on the patient context provided above, answer the doctor's question
     messages.append({"role": "user", "content": [{"type": "input_text", "text": message}]})
 
     try:
+        # Format the messages for GPT-4.1 API
+        system_prompt = messages[0]["content"]
+        # Combine all subsequent messages into the input
+        user_messages = []
+        for msg in messages[1:]:
+            if msg["role"] == "user":
+                if isinstance(msg["content"], list):
+                    for content in msg["content"]:
+                        if isinstance(content, dict) and content.get("type") == "input_text":
+                            user_messages.append(content["text"])
+                        else:
+                            user_messages.append(str(msg["content"]))
+                else:
+                    user_messages.append(msg["content"])
+        
+        combined_input = "\n\n".join(user_messages)
+        
         chat_completion = openai_client.responses.create(
             model="gpt-4.1",
-            input=messages,
-            max_tokens=1024
+            instructions=system_prompt,
+            input=combined_input
         )
 
         response = chat_completion.output_text