Update app.py

app.py

@@ -40,11 +40,11 @@ def predict_image(input_image, name, patient_id):
 
             # Determine the label based on the class index
             if class_index == 0:
-                label = "Mature"
-                color = (255, 0, 0)  # Red for Mature
-            elif class_index == 1:
                 label = "Immature"
                 color = (0, 255, 255)  # Yellow for Immature
+            elif class_index == 1:
+                label = "Mature"
+                color = (255, 0, 0)  # Red for Mature
             else:
                 label = "Normal"
                 color = (0, 255, 0)  # Green for Normal
@@ -238,24 +238,25 @@ with app:
         with gr.Column():
             name = gr.Textbox(label="Name")
             patient_id = gr.Textbox(label="Patient ID")
-            input_image = gr.Image(type="pil", label="Upload an Image", image_mode="RGB")
-
-            with gr.Row():
-                download_choice = gr.Radio(label="Download Choice", choices=["Database (.db)", "Last Predicted Image (.png)"], value="Database (.db)")
-                download_button = gr.Button("Download")
+            input_image = gr.Image(label="Input Image", tool="editor", type="pil")
 
         with gr.Column():
-            predicted_image = gr.Image(label="Predicted Image")
-            raw_result = gr.Textbox(label="Raw Prediction Results", interactive=False)
-            submit_status = gr.Textbox(label="Submission Status", interactive=False)
+            output_image = gr.Image(label="Predicted Image")
+            raw_result = gr.Textbox(label="Raw Predictions", lines=5)
+            submit_status = gr.Textbox(label="Submit Status")
+
+        predict_button = gr.Button("Predict")
+
+        predict_button.click(fn=interface, inputs=[name, patient_id, input_image], outputs=[output_image, raw_result, submit_status])
 
-    predict_button = gr.Button("Predict")
-    view_db_button = gr.Button("View Database")
+    with gr.Row():
+        with gr.Column():
+            view_button = gr.Button("View Database")
+            download_choice = gr.Dropdown(label="Download Option", choices=["Database (.db)", "Predicted Image (.png)"])
+            download_button = gr.Button("Download")
 
-    # Button actions
-    predict_button.click(interface, inputs=[name, patient_id, input_image], outputs=[predicted_image, raw_result, submit_status])
-    view_db_button.click(view_db_interface, outputs=gr.Dataframe())
-    download_button.click(download_interface, inputs=[download_choice], outputs=[gr.File(), gr.Textbox()])
+            view_button.click(fn=view_db_interface, inputs=[], outputs=[gr.Dataframe()])
+            download_button.click(fn=download_interface, inputs=[download_choice], outputs=[gr.File(), gr.Textbox()])
 
 # Launch the Gradio app
 app.launch()