Update tests/test_gradio.py

tests/test_gradio.py

@@ -1,33 +1,40 @@
 import pytest
-import sys
-import os
+import gradio as gr
+from app import process_image, initialize_model
 import numpy as np
 from PIL import Image
 
-# Add the parent directory to the path to import app
-sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-
-def test_gradio_interface():
-    """Test Gradio interface components"""
-    # Test model initialization simulation
-    def mock_initialize_model():
-        return "Model initialized successfully", None, None, None
+def test_initialize_model():
+    """Test model initialization through Gradio interface"""
+    result = initialize_model()
+    assert isinstance(result, tuple), "Initialize model should return a tuple"
+    assert len(result) == 4, "Initialize model should return 4 values"
     
-    result = mock_initialize_model()
-    assert "successfully" in result[0].lower(), "Model initialization failed"
+    status_message = result[0]
+    assert isinstance(status_message, str), "Status message should be a string"
 
-def test_process_image():
-    """Test image processing functionality"""
+def test_process_image_no_model():
+    """Test image processing without initialized model"""
+    # Reset global model to None
+    import app
+    app.yolo_model = None
+    
     # Create a dummy image
     image = Image.fromarray(np.zeros((640, 640, 3), dtype=np.uint8))
+    result = process_image(image)
+    
+    assert isinstance(result, tuple), "Process image should return a tuple"
+    assert len(result) == 4, "Process image should return 4 values"
+    
+    output_image, output_text, pdf_file, violations = result
+    assert "not initialized" in output_text.lower(), "Should indicate model not initialized"
+
+def test_process_image_no_input():
+    """Test image processing with no image input"""
+    result = process_image(None)
     
-    # Mock the process_image function behavior
-    def mock_process_image(img):
-        if img is not None:
-            return img, "Image processed successfully. No violations detected. Processing time: 0.1 seconds.", None, []
-        else:
-            return None, "Error: No image provided", None, []
+    assert isinstance(result, tuple), "Process image should return a tuple"
+    assert len(result) == 4, "Process image should return 4 values"
     
-    output_image, output_text, pdf_output, violations = mock_process_image(image)
-    assert output_text is not None, "Process image returned None"
-    assert "processed successfully" in output_text.lower(), "Image processing failed"
+    output_image, output_text, pdf_file, violations = result
+    assert "no image" in output_text.lower(), "Should indicate no image provided"