Update import paths in test_inference_pipeline.py for phishing detection model

app.py

@@ -54,6 +54,44 @@ def initialize_app():
         return False
 
 
+def get_model_info() -> str:
+    """Get model information as HTML."""
+    if not pipeline:
+        return "<p>Model information not available</p>"
+
+    # TODO: Check if pipeline has model_metrics attribute
+    metrics = getattr(pipeline, 'model_metrics', None)
+    if not metrics:
+        return "<p>Model metrics not available</p>"
+
+    version = getattr(pipeline, 'model_version', 'latest')
+
+    html = f"""
+    <div style="padding: 20px; background-color: #f5f5f5; border-radius: 10px; margin: 10px;">
+        <h3>Model Information</h3>
+        <p><strong>Model Name:</strong> {pipeline.model_name}</p>
+        <p><strong>Model Version:</strong> {version}</p>
+        <hr>
+        <h4>Performance Metrics</h4>
+        <div style="display: grid; grid-template-columns: 1fr 1fr; gap: 10px;">
+            <div><strong>Test Accuracy:</strong> {metrics.get('test_accuracy', 'N/A'):.4f}</div>
+            <div><strong>Test Precision:</strong> {metrics.get('test_precision', 'N/A'):.4f}</div>
+            <div><strong>Test Recall:</strong> {metrics.get('test_recall', 'N/A'):.4f}</div>
+            <div><strong>Test F1 Score:</strong> {metrics.get('test_f1_score', 'N/A'):.4f}</div>
+            <div><strong>Test ROC-AUC:</strong> {metrics.get('test_roc_auc', 'N/A'):.4f}</div>
+            <div><strong>Validation F1 Score:</strong> {metrics.get('val_f1_score', 'N/A'):.4f}</div>
+        </div>
+        <hr>
+        <h4>Training Information</h4>
+        <p><strong>Number of Features:</strong> {metrics.get('n_features', 'N/A')}</p>
+        <p><strong>Training Samples:</strong> {metrics.get('n_train_samples', 'N/A')}</p>
+        <p><strong>Test Samples:</strong> {metrics.get('n_test_samples', 'N/A')}</p>
+        <p><strong>Best CV Score:</strong> {metrics.get('best_cv_score', 'N/A'):.4f}</p>
+    </div>
+    """
+    return html
+
+
 def gradio_interface(url: str) -> Tuple[str, str, str]:
     """
     Gradio interface function.
@@ -151,42 +189,80 @@ def create_gradio_app():
             """
         )
 
-        with gr.Row():
-            with gr.Column(scale=3):
-                url_input = gr.Textbox(
-                    label="URL to Check",
-                    placeholder="Enter URL (e.g., example.com or https://example.com)",
-                    lines=1
+        # Create tabs for URL Checker and Model Info
+        with gr.Tabs():
+            # Tab 1: URL Checker
+            with gr.Tab("🔍 URL Checker"):
+                with gr.Row():
+                    with gr.Column(scale=3):
+                        url_input = gr.Textbox(
+                            label="URL to Check",
+                            placeholder="Enter URL (e.g., example.com or https://example.com)",
+                            lines=1
+                        )
+                    with gr.Column(scale=1):
+                        submit_btn = gr.Button("Check URL", variant="primary", size="lg")
+
+                with gr.Row():
+                    result_output = gr.HTML(label="Prediction")
+
+                with gr.Row():
+                    confidence_output = gr.HTML(label="Confidence")
+
+                with gr.Row():
+                    details_output = gr.HTML(label="Details")
+
+                # Example URLs
+                gr.Markdown("### Example URLs to Try:")
+                gr.Examples(
+                    examples=[
+                        ["https://google.com"],
+                        ["https://github.com"],
+                        ["https://facebook.com"],
+                    ],
+                    inputs=url_input,
                 )
-            with gr.Column(scale=1):
-                submit_btn = gr.Button("Check URL", variant="primary", size="lg")
-
-        with gr.Row():
-            result_output = gr.HTML(label="Prediction")
-
-        with gr.Row():
-            confidence_output = gr.HTML(label="Confidence")
-
-        with gr.Row():
-            details_output = gr.HTML(label="Details")
-
-        # Example URLs
-        gr.Markdown("### Example URLs to Try:")
-        gr.Examples(
-            examples=[
-                ["https://google.com"],
-                ["https://github.com"],
-                ["https://facebook.com"],
-            ],
-            inputs=url_input,
-        )
 
-        # Connect button to function
-        submit_btn.click(
-            fn=gradio_interface,
-            inputs=url_input,
-            outputs=[result_output, confidence_output, details_output]
-        )
+                # Connect button to function
+                submit_btn.click(
+                    fn=gradio_interface,
+                    inputs=url_input,
+                    outputs=[result_output, confidence_output, details_output]
+                )
+
+            # Tab 2: Model Information
+            with gr.Tab("📊 Model Info"):
+                gr.HTML(value=get_model_info(), label="Model Statistics")
+
+                gr.Markdown("### Model Evaluation Visualizations")
+
+                # TODO: Check if pipeline has model_dir attribute to access images
+                model_dir = getattr(pipeline, 'model_dir', None) if pipeline else None
+
+                with gr.Row():
+                    with gr.Column():
+                        gr.Markdown("#### Confusion Matrix")
+                        # TODO: Verify the image filename in Hopsworks model artifacts
+                        if model_dir:
+                            confusion_path = os.path.join(model_dir, "confusion_matrices.png")
+                            if os.path.exists(confusion_path):
+                                gr.Image(value=confusion_path, label="Confusion Matrix")
+                            else:
+                                gr.Markdown("*Confusion matrix image not found*")
+                        else:
+                            gr.Markdown("*Model directory not available*")
+
+                    with gr.Column():
+                        gr.Markdown("#### Feature Importance")
+                        # TODO: Verify the image filename in Hopsworks model artifacts
+                        if model_dir:
+                            importance_path = os.path.join(model_dir, "feature_importance.png")
+                            if os.path.exists(importance_path):
+                                gr.Image(value=importance_path, label="Feature Importance")
+                            else:
+                                gr.Markdown("*Feature importance image not found*")
+                        else:
+                            gr.Markdown("*Model directory not available*")
 
         gr.Markdown(
             """