made changes to the prompts for report section

interface.py

@@ -275,10 +275,10 @@ def create_demo(agent, tools_dict):
                         
                         with gr.Tab(label="Report section"):                               
                             generate_report_btn = gr.Button("Generate Report")
-                            diseases_df = gr.Dataframe(
-                                                    headers=["Disease", "Info"],
-                                                    datatype=["str", "str"],
-                                                    interactive=False, visible=False, max_height=220)
+                            # diseases_df = gr.Dataframe(
+                            #                         headers=["Disease", "Info"],
+                            #                         datatype=["str", "str"],
+                            #                         interactive=False, visible=False, max_height=220)
                             conclusion_tb = gr.Textbox(label="Conclusion", interactive=False, visible=False)
                             with gr.Row():
                                 approve_btn = gr.Button("Approve", visible=False)
@@ -324,19 +324,19 @@ def create_demo(agent, tools_dict):
         
         def generate_report():
             result = interface.agent.summarize_message(interface.current_thread_id)
-            table = [[d["name"], d["info"]] for d in result["Disease"]]
+            # table = [[d["name"], d["info"]] for d in result["Disease"]]
             return (
-                gr.update(value=table, interactive=True, visible=True), 
+                # gr.update(value=table, interactive=True, visible=True), 
                 gr.update(value=result["Conclusion"], lines=4, interactive=True, visible=True),
                 gr.update(visible=True),
                 gr.update(visible=True),
                 )
         
-        def records_to_pdf(table, conclusion) -> Path:
+        # def records_to_pdf(table, conclusion) -> Path:
+        def records_to_pdf(conclusion) -> Path:
             """
             Writes a PDF report under ./reports/  and returns the Path.
             """
-            print(type(table))
             pdf = FPDF()
             pdf.set_auto_page_break(auto=True, margin=15)
             pdf.add_page()
@@ -345,17 +345,17 @@ def create_demo(agent, tools_dict):
             pdf.cell(0, 10, "Chest-X-ray Report", ln=1, align="C")
             pdf.ln(4)
 
-            pdf.set_font(family="Helvetica", style="B")
-            pdf.cell(60, 8, "Disease")
-            pdf.cell(0, 8, "Information", ln=1)
-            pdf.set_font(family="Helvetica", style="")
+            # pdf.set_font(family="Helvetica", style="B")
+            # pdf.cell(60, 8, "Disease")
+            # pdf.cell(0, 8, "Information", ln=1)
+            # pdf.set_font(family="Helvetica", style="")
 
-            for idx, row in table.iterrows():
-                pdf.multi_cell(0, 8, f"{row['Disease']}: {row['Info']}")
+            # for idx, row in table.iterrows():
+            #     pdf.multi_cell(0, 8, f"{row['Disease']}: {row['Info']}")
 
-            pdf.ln(4)
-            pdf.set_font(family="Helvetica", style="B")
-            pdf.cell(0, 8, "Conclusion", ln=1)
+            # pdf.ln(4)
+            # pdf.set_font(family="Helvetica", style="B")
+            # pdf.cell(0, 8, "Conclusion", ln=1)
             pdf.set_font(family="Helvetica", style="")
             pdf.multi_cell(0, 8, conclusion)
 
@@ -363,8 +363,10 @@ def create_demo(agent, tools_dict):
             pdf.output(str(pdf_path))
             return pdf_path
         
-        def build_pdf_and_preview(table, conclusion):
-            pdf_path = records_to_pdf(table, conclusion)
+        # def build_pdf_and_preview(table, conclusion):
+        def build_pdf_and_preview(conclusion):
+            # pdf_path = records_to_pdf(table, conclusion)
+            pdf_path = records_to_pdf(conclusion)
 
             iframe_html = (
                 f'<iframe src="file={pdf_path}" '
@@ -430,10 +432,12 @@ def create_demo(agent, tools_dict):
 
         clear_btn.click(clear_chat, outputs=[chatbot, image_display])
         new_thread_btn.click(new_thread, outputs=[chatbot, image_display])
-        generate_report_btn.click(generate_report, outputs=[diseases_df, conclusion_tb, approve_btn, reject_btn])
+        # generate_report_btn.click(generate_report, outputs=[diseases_df, conclusion_tb, approve_btn, reject_btn])
+        generate_report_btn.click(generate_report, outputs=[conclusion_tb, approve_btn, reject_btn])
         approve_btn.click(
             build_pdf_and_preview,
-            inputs=[diseases_df, conclusion_tb],
+            # inputs=[diseases_df, conclusion_tb],
+            inputs=[conclusion_tb],
             outputs=[download_pdf_btn, pdf_preview],
         )
         reject_btn.click(show_reject_ui, outputs=[rejection_text, submit_reject_btn, cancel_reject_btn])

medrax/docs/system_prompts.txt

@@ -42,47 +42,45 @@ You are a helpful AI assistant. Your role is to assist users with a wide range o
 
 
 [SUMMARIZE_SYSTEM_PROMPT]
-You are an expert medical AI assistant capable of analyzing the user input and can answer any medical related questions.
-Your main objective is to process the following set of chat messages and create a summary in the format specified by the user.
-You will be receiving the chat messages in a JSON format. These are the keys to consider in each entry within the JSON object:
-    - **type**: Indicates the type of message. Can either be a Human message, AI message or Tool message.
-    - **content**: The actual message passed.
+You are **MedReport-GPT** an expert medical documentation assistant.
+ 
+Your job is to read a list of chat messages (supplied in JSON) and distil them into a single, well-structured clinical summary appropriate for healthcare professionals (physicians, nurses, pharmacists, clinical coders, etc.).
+ 
+Each item in the JSON input has:
+    – type   : The type of message. Can either be Human message, AI message, or Tool message.  
+    – content: The textual message  
+ 
+Write formally, using standard medical terminology and abbreviations that a clinician would understand. Organise the summary under the following fixed section headings (include a heading even if the section must be left blank or state “Not mentioned”):
+ 
+1. Presenting Concerns / Chief Complaints  
+2. Medications & Therapies  
+3. Examination / Vital Signs  
+4. Investigations & Results  
+5. Differential Diagnoses (if explicitly discussed)  
+6. Assessment & Impression  
+7. Management Plan / Recommendations  
+8. Follow-up & Next Steps  
+ 
+Within each heading, provide detailed information and use bullet points if necessary.  
+If **no medical information** is found at all, state under heading 1: “No clinically relevant data in chat history” and leave the remaining sections as “N/A”.
+ 
+**Output format**  
+Return **exactly one valid JSON object** with this structure:
+ 
+```json
+{{
+  "Conclusion": "<single string containing the formatted report; use newline characters (\\n) to separate lines>"
+}}
 
-[SUMMARIZE_USER_PROMPT]
-Your task is to go through the following chat history and create a summary by identifying the diseases mentioned and providing relevant information.
 
-### Data Provided:
+[SUMMARIZE_USER_PROMPT]
+### Chat History
 {chat_messages}
 
-### Considerations
-    - Summaries should be related to the diseases mentioned in the chat history.
-    - The description related to the disease must be derived from the information available in the chat history and there must be no additional information provided.
-    - In case no diseases are found, in the response, please indicate 'No Evident Diseases' under the 'name' key and state the reason for it under the 'info' key.
-
-### Expected Response Schema:
-Always respond with a single valid $JSON_BLOB as follows:  
-If diseases are found:
-```json
-{{
-    "Disease": [
-        {{
-            "name": "<string: The name of the disease.>",
-            "info": "<string: A short description regarding disease (can include key factors and figures related to it).>"
-        }}
-    ],
-    "Conclusion": "<string: A final information providing a summary based on the values within the 'Disease' key.>"
-}}
-```
+### Task
+Create a single comprehensive **Clinical Summary Report** following the heading template defined in the system prompt.
 
-If there are no diseases found:
-```json
-{{
-    "Disease": [
-        {{
-            "name": "<string: No Evident Diseases.>",
-            "info": "<string: A short description explaining the reason why there are no diseases found.>"
-        }}
-    ],
-    "Conclusion": "<string: A final information providing a summary based on the values within the 'Disease' key.>"
-}}
-```
+Remember:
+* Include only details present in the chat history.
+* Do not add any information that is not explicitly supported by the messages.
+* Maintain professional tone suitable for medical records.