Add enhanced logging for debugging nemaquant.py execution

app.py

@@ -91,11 +91,22 @@ def process_images():
 
     try:
         print(f"Running command: {' '.join(cmd)}") # Log the command
+        print(f"Input directory contents: {os.listdir(str(job_input_dir))}")
+        print(f"Weights file exists: {os.path.exists(str(WEIGHTS_FILE))}")
+        
         # Run the script, capture output and errors
         # Timeout might be needed for long processes on shared infrastructure like HF Spaces
         result = subprocess.run(cmd, capture_output=True, text=True, check=True, timeout=300) # 5 min timeout
         status_log = f"NemaQuant Output:\n{result.stdout}\nNemaQuant Errors:\n{result.stderr}"
         print(status_log) # Log script output
+        
+        # Check output directory after command runs
+        print(f"Output directory exists: {os.path.exists(str(job_output_dir))}")
+        if os.path.exists(str(job_output_dir)):
+            print(f"Output directory contents: {os.listdir(str(job_output_dir))}")
+        
+        # Check output file
+        print(f"Output CSV exists: {os.path.exists(str(output_csv))}")
 
         # --- Parse Results ---
         if not output_csv.exists():
@@ -113,6 +124,8 @@ def process_images():
             suffix = Path(original_filename).suffix
             annotated_filename = f"{stem}_annotated{suffix}"
             annotated_path = annotated_dir / annotated_filename
+            
+            print(f"Looking for annotated file: {annotated_path}, exists: {annotated_path.exists()}")
 
             results_list.append({
                 "filename": original_filename,
@@ -142,7 +155,9 @@ def process_images():
         print(error_message)
         return jsonify({"error": "Could not find output file", "log": error_message}), 500
     except Exception as e:
-        error_message = f"An unexpected error occurred: {str(e)}"
+        error_message = f"An unexpected error occurred: {str(e)}\n"
+        import traceback
+        error_message += traceback.format_exc()
         print(error_message)
         return jsonify({"error": "An unexpected error occurred", "log": error_message}), 500