Update app.py

app.py

@@ -84,7 +84,7 @@ def preprocess_image(image):
 def predict_xray(image):
     try:
         if image is None:
-            return "Please upload an image.", "Please upload an image."
+            return "Please upload an image."
 
         img_tensor = preprocess_image(image)
         with torch.no_grad():
@@ -103,10 +103,6 @@ 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>
             """
 
@@ -117,22 +113,17 @@ 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>
-            <p><b>Recommendation:</b> {info['recommendation']}</p>
         </div>
         """
 
     except Exception as e:
         logger.error(f"Error in prediction: {e}")
-        return f"Error: {str(e)}", "Error in prediction"
+        return f"Error: {str(e)}"
 
 # Analyze PDF report
 def analyze_report(file):
     if not file or not file.name.endswith(".pdf"):
-        return "Please upload a valid PDF file.", "Please upload a valid PDF file."
+        return "Please upload a valid PDF file."
     try:
         doc = fitz.open(file.name)
         text = "".join(page.get_text() for page in doc)
@@ -148,10 +139,10 @@ 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}, <h4>Summary:</h4><p><b>Condition:</b> {condition}</p><p><b>Status:</b> {status}</p>", f"Condition: {condition}, Status: {status}"
+        return f"Condition: {condition}\nDisease: {disease}\nStatus: {status}\n\nPreview:\n{preview}"
 
     except Exception as e:
-        return f"Failed to process PDF: {str(e)}", "Failed to process PDF"
+        return f"Failed to process PDF: {str(e)}"
 
 # Gradio interface
 def create_interface():
@@ -160,17 +151,14 @@ def create_interface():
 
         with gr.Tabs():
             with gr.TabItem("X-ray Analysis"):
-                with gr.Row():
-                    img_input = gr.Image(label="Upload Chest X-ray", type="pil")
-                    img_output = gr.HTML()
-                    summary_output = gr.HTML(label="Summary Result")
-                gr.Button("Analyze X-ray", elem_id="analyze_button", scale=0.5).click(predict_xray, inputs=img_input, outputs=[img_output, summary_output])
+                img_input = gr.Image(label="Upload Chest X-ray", type="pil")
+                summary_output = gr.HTML(label="Summary Result")
+                gr.Button("Analyze X-ray", elem_id="analyze_button", scale=0.5).click(predict_xray, inputs=img_input, outputs=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)
                 summary_output_report = gr.Textbox(label="Summary Result", lines=5)
-                gr.Button("Analyze Report", elem_id="analyze_button", scale=0.5).click(analyze_report, inputs=pdf_input, outputs=[pdf_output, summary_output_report])
+                gr.Button("Analyze Report", elem_id="analyze_button", scale=0.5).click(analyze_report, inputs=pdf_input, outputs=summary_output_report)
 
     return demo