Update app.py

app.py

@@ -548,10 +548,10 @@ def classical_prediction(file_path, model_choice, min_mmsi, max_mmsi, models, lo
             positions_csv_path = tmp_positions_file.name
 
         logging.info("Classical prediction completed.")
-        return metrics_json, positions_csv_path, inference_time
+        return metrics_json, positions_csv_path, inference_time, None
     except Exception as e:
         logging.error(f"An error occurred: {str(e)}")
-        return {"error": str(e)}, None, None
+        return None, None, None, str(e)
 
 # ============================
 # Abnormal Behavior Detection
@@ -650,10 +650,10 @@ def abnormal_behavior_detection(prediction_file_path, alpha=0.5, threshold=10.0)
             abnormal_csv_path = tmp_abnormal_file.name
 
         logging.info("Abnormal behavior detection completed.")
-        return abnormal_csv_path
+        return abnormal_csv_path, None
     except Exception as e:
         logging.error(f"An error occurred: {str(e)}")
-        return {"error": str(e)}
+        return None, str(e)
 
 # ============================
 # Define Gradio Interface
@@ -698,8 +698,9 @@ def main():
         ],
         outputs=[
             gr.JSON(label="Classical Metrics (Degrees)"),
-            gr.File(label="Download Predicted & Real Positions CSV", optional=True),
-            gr.Number(label="Inference Time (seconds)")
+            gr.File(label="Download Predicted & Real Positions CSV"),
+            gr.Number(label="Inference Time (seconds)"),
+            gr.Textbox(label="Error Message", lines=2, visible=False)
         ],
         title="Classical Prediction & Metrics",
         description="Upload a CSV file and select a model to get classical evaluation metrics such as MAE, MSE, RMSE. The inference time is also provided."
@@ -714,7 +715,8 @@ def main():
             gr.Number(label="Threshold", value=10.0)
         ],
         outputs=[
-            gr.File(label="Download Abnormal Behavior CSV", optional=True)
+            gr.File(label="Download Abnormal Behavior CSV"),
+            gr.Textbox(label="Error Message", lines=2, visible=False)
         ],
         title="Abnormal Behavior Detection",
         description=(