Update app.py

app.py

@@ -71,6 +71,7 @@ def unzip_and_find_jpgs(file_path):
     return jpg_files
 
 def process_images(uploaded_file):
+    logging.info("Started processing the uploaded file.")  # Check if the function is triggered
     global results_df
     results_df = results_df.iloc[0:0]  # Clear the DataFrame
     file_path = uploaded_file.name  # Gradio provides the file path
@@ -80,9 +81,11 @@ def process_images(uploaded_file):
         image_files = unzip_and_find_jpgs(file_path)
         
         if not image_files:
+            logging.warning("No JPG files found in the zip.")
             return "No JPG files found in the zip."
 
         for file_path in image_files:
+            logging.info(f"Processing image file {file_path}.")
             extracted_text, translated_text = batch_process_documents(file_path, "image/jpeg")
             new_row = pd.DataFrame([{
                 "Filename": os.path.basename(file_path),
@@ -90,6 +93,7 @@ def process_images(uploaded_file):
                 "Translated Text": translated_text
             }])
             results_df = pd.concat([results_df, new_row], ignore_index=True)
+            logging.info(f"Data added for file {file_path}.")
     except Exception as e:
         logging.error(f"An error occurred: {str(e)}")
         return f"An error occurred: {str(e)}"
@@ -99,7 +103,7 @@ def process_images(uploaded_file):
 
 interface = gr.Interface(
     fn=process_images,
-    inputs="file",
+    inputs=gr.inputs.File(label="Upload ZIP File"),
     outputs="html",
     title="Document AI Translation",
     description="Upload a ZIP file containing JPEG/JPG images, and the system will extract and translate text from each image."
@@ -107,3 +111,4 @@ interface = gr.Interface(
 
 if __name__ == "__main__":
     interface.launch(debug=True)
+