Update app.py

app.py

@@ -21,20 +21,26 @@ def generate_caption(image):
     if any(keyword in caption.lower() for keyword in dal_keywords):
         return f"The image is related to dal"
     else:
-        return ("This image is not related to dal")
+        return None  # Return None if the image is not related to dal
 
 # Define the function for the Gradio interface
 def captioning_interface(image):
-    return generate_caption(image)
+    # Generate caption and check if related to dal
+    result = generate_caption(image)
+    if result:
+        return image, result  # If related to dal, show the image and the message
+    else:
+        raise gr.Error("This image is not related to dal. Please upload an image related to dal.")  # Trigger error
 
 # Create the Gradio interface
 interface = gr.Interface(
     fn=captioning_interface,
-    inputs=gr.Image(type="pil"),
-    outputs=gr.Textbox(),
+    inputs=gr.Image(type="pil", label="Upload Image", image_mode='RGB'),  # Removed the 'tool' argument
+    outputs=[gr.Image(type="pil", label="Image Preview"), gr.Textbox(label="Description")],  # Show image only if related to dal
     title="Dal Detection",
-    description="Upload an image to get a textual description. Only images related to dal (lentil) will be accepted."
+    description="Upload an image to get a textual description. Only images related to dal (lentil) will be accepted.",
+    allow_flagging="never"  # Disable flagging for rejected images
 )
 
 # Launch the interface
-interface.launch(share=True)
+interface.launch(share=True)