Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-from gradio.components import Image, Textbox, Markdown
 import cv2
 import numpy as np
 from groq import Groq
@@ -72,7 +71,7 @@ def create_monitor_interface():
 
         def process_frame(self, frame: np.ndarray) -> tuple[np.ndarray, str]:
             if frame is None:
-                return None, "No frame received"
+                return None, "No image provided"
                 
             analysis = self.analyze_frame(frame)
             display_frame = frame.copy()
@@ -94,13 +93,16 @@ def create_monitor_interface():
     monitor = SafetyMonitor()
     
     with gr.Blocks() as demo:
-        gr.Markdown("# Real-time Safety Monitoring System")
+        gr.Markdown("""
+        # Real-time Safety Monitoring System
+        Upload an image to analyze workplace safety concerns.
+        """)
         
         with gr.Row():
-            input_image = Image(label="Input Image", type="numpy", tool="upload")
-            output_image = Image(label="Analysis Feed", type="numpy")
+            input_image = gr.Image(label="Upload Image")
+            output_image = gr.Image(label="Analysis Results")
         
-        analysis_text = Textbox(label="Safety Analysis", lines=5)
+        analysis_text = gr.Textbox(label="Safety Analysis", lines=5)
             
         def analyze_image(image):
             if image is None:
@@ -110,15 +112,15 @@ def create_monitor_interface():
             
         input_image.change(
             fn=analyze_image,
-            inputs=[input_image],
-            outputs=[output_image, analysis_text],
+            inputs=input_image,
+            outputs=[output_image, analysis_text]
         )
         
         gr.Markdown("""
         ## Instructions:
-        1. Upload an image using the input panel
-        2. The system will automatically analyze the image for safety concerns
-        3. Results will be shown in the analysis feed and text box
+        1. Click the 'Upload Image' area or drag and drop an image
+        2. The system will automatically analyze the safety concerns
+        3. View the analyzed image and detailed safety report
         """)
 
     return demo