Update app.py

app.py

@@ -8,6 +8,10 @@ from PIL import Image
 import requests
 import os
 from io import BytesIO
+import logging
+
+# Set up basic logging
+logging.basicConfig(level=logging.INFO)
 
 # Define the number of classes
 num_classes = 2
@@ -66,17 +70,25 @@ def predict_from_image(image):
 def process_file_path(file_path_input):
     global file_path
     file_path = file_path_input  # Store the file path
+    
+    logging.info(f"Received file path: {file_path}")
+    
     if not os.path.exists(file_path):
+        logging.error(f"File not found at {file_path}")
         return {"error": f"File not found at {file_path}"}
+    
     image = Image.open(file_path)
+    logging.info(f"Processing image from path: {file_path}")
     return predict_from_image(image)
 
 # Function to fetch the result (for the GET request)
 def fetch_result():
     if file_path:
         image = Image.open(file_path)
+        logging.info(f"Making prediction for image at path: {file_path}")
         return predict_from_image(image)
     else:
+        logging.warning("No file path available. Please send a POST request with a file path first.")
         return {"error": "No file path available. Please send a POST request with a file path first."}
 
 # Gradio interface