Update app.py

app.py

@@ -103,6 +103,10 @@ def predict_xray(image):
                 <p><b>Confidence:</b> {confidence:.2f}%</p>
                 <p><b>Note:</b> The model is not confident enough to provide a clear diagnosis.</p>
                 <p><b>Recommendation:</b> Please consult a radiologist or upload a better-quality image.</p>
+            </div>, <div style="font-family:Arial">
+                <h4>Summary:</h4>
+                <p><b>Diagnosis:</b> Uncertain</p>
+                <p><b>Confidence Level:</b> {confidence:.2f}%</p>
             </div>
             """
 
@@ -113,6 +117,10 @@ def predict_xray(image):
             <p><b>Confidence:</b> {confidence:.2f}%</p>
             <p><b>Description:</b> {info['description']}</p>
             <p><b>Recommendation:</b> {info['recommendation']}</p>
+        </div>, <div style="font-family:Arial">
+            <h4>Summary:</h4>
+            <p><b>Diagnosis:</b> {top_condition}</p>
+            <p><b>Confidence Level:</b> {confidence:.2f}%</p>
         </div>
         """
 
@@ -139,7 +147,7 @@ def analyze_report(file):
             condition, disease, status = "Fracture", "Bone Injury", "Orthopedic Attention Required"
 
         preview = text[:300] + "..." if text else "No readable content."
-        return f"Condition: {condition}\nDisease: {disease}\nStatus: {status}\n\nPreview:\n{preview}"
+        return f"Condition: {condition}\nDisease: {disease}\nStatus: {status}\n\nPreview:\n{preview}, <h4>Summary:</h4><p><b>Condition:</b> {condition}</p><p><b>Status:</b> {status}</p>"
 
     except Exception as e:
         return f"Failed to process PDF: {str(e)}"
@@ -153,12 +161,14 @@ def create_interface():
             with gr.TabItem("X-ray Analysis"):
                 img_input = gr.Image(label="Upload Chest X-ray", type="pil")
                 img_output = gr.HTML()
-                gr.Button("Analyze X-ray").click(predict_xray, inputs=img_input, outputs=img_output)
+                summary_output = gr.HTML(label="Summary Result")
+                gr.Button("Analyze X-ray").click(predict_xray, inputs=img_input, outputs=[img_output, summary_output])
 
             with gr.TabItem("Report Analysis"):
                 pdf_input = gr.File(label="Upload PDF Report", file_types=[".pdf"])
                 pdf_output = gr.Textbox(label="Extracted Summary", lines=10)
-                gr.Button("Analyze Report").click(analyze_report, inputs=pdf_input, outputs=pdf_output)
+                summary_output_report = gr.Textbox(label="Summary Result", lines=5)
+                gr.Button("Analyze Report").click(analyze_report, inputs=pdf_input, outputs=[pdf_output, summary_output_report])
 
     return demo