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."
+            return "Please upload an image.", "Please upload an image."
 
         img_tensor = preprocess_image(image)
         with torch.no_grad():
@@ -126,12 +126,12 @@ def predict_xray(image):
 
     except Exception as e:
         logger.error(f"Error in prediction: {e}")
-        return f"Error: {str(e)}"
+        return f"Error: {str(e)}", "Error in prediction"
 
 # Analyze PDF report
 def analyze_report(file):
     if not file or not file.name.endswith(".pdf"):
-        return "Please upload a valid PDF file."
+        return "Please upload a valid PDF file.", "Please upload a valid PDF file."
     try:
         doc = fitz.open(file.name)
         text = "".join(page.get_text() for page in doc)
@@ -147,10 +147,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>"
+        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}"
 
     except Exception as e:
-        return f"Failed to process PDF: {str(e)}"
+        return f"Failed to process PDF: {str(e)}", "Failed to process PDF"
 
 # Gradio interface
 def create_interface():
@@ -159,9 +159,10 @@ def create_interface():
 
         with gr.Tabs():
             with gr.TabItem("X-ray Analysis"):
-                img_input = gr.Image(label="Upload Chest X-ray", type="pil")
-                img_output = gr.HTML()
-                summary_output = gr.HTML(label="Summary Result")
+                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").click(predict_xray, inputs=img_input, outputs=[img_output, summary_output])
 
             with gr.TabItem("Report Analysis"):