Update get_answer.py

get_answer.py

@@ -11,7 +11,7 @@ def encode_image(image_path):
 
 
 client = OpenAI(api_key=openai_api)
-def get_ai_response(prompt_content):
+def get_ai_response(prompt_content, prompt):
     response = client.chat.completions.create(
     model="gpt-4o",
     messages=[
@@ -20,25 +20,7 @@ def get_ai_response(prompt_content):
         "content": [
             {
             "type": "text",
-            "text": f"""You are an experimental AI-copilot for doctors. They will check your outputs
-            You will be given by the doctor patient data input and your role will be to determine the most probable diagnose.
-            You will include all relevant literature backup and references needed and a whole reasoning path of why you think it is.
-            Be very professional and redact as a health practioner.
-            
-            You format each output with only:
-            
-            # Probable diagnose: X
-            
-            #### likelihood: XX% 
-            
-            # Suggested treatment:
-              -XX
-              - XX
-              
-            # Full reasoning and path to diagnose (as a clean decision path with all relevant elements) 
-             
-            
-            """
+            "text": prompt
             }
         ]
         },
@@ -52,12 +34,11 @@ def get_ai_response(prompt_content):
     top_p=1,
     frequency_penalty=0,
     presence_penalty=0,
-    stream=True
     )
-    return response
+    return response.choices[0].message.content
 
 
-def get_answer(patient_data):
-    answer = get_ai_response(patient_data)
+def get_answer(patient_data, prompt):
+    answer = get_ai_response(patient_data, prompt)
     # answer = ""
     return answer