Update app.py

app.py

@@ -9,28 +9,35 @@ from io import BytesIO
 from datasets.filesystems import S3FileSystem
 import s3fs
 
+
 # AWS Setup
 aws_access_key_id = os.getenv('AWS_ACCESS_KEY')
 aws_secret_access_key = os.getenv('AWS_SECRET_KEY')
 region_name = os.getenv('AWS_REGION')
 s3_bucket = os.getenv('AWS_BUCKET')
 
+# Initialize s3fs with your AWS credentials
+s3_fs = s3fs.S3FileSystem(key=aws_access_key_id, secret=aws_secret_access_key, client_kwargs={'region_name': region_name})
+
 textract_client = boto3.client('textract', aws_access_key_id=aws_access_key_id, aws_secret_access_key=aws_secret_access_key, region_name=region_name)
-s3_client = boto3.client('s3', aws_access_key_id=aws_access_key_id, aws_secret_access_key=aws_secret_access_key, region_name=region_name)
 
 def upload_file_to_s3(file_content, bucket, object_name=None):
+    """Uploads file to S3 using s3fs."""
     if object_name is None:
-        object_name = os.path.basename(file_content)
+        raise ValueError("object_name cannot be None")
+    
     try:
-        s3_client.upload_fileobj(file_content, bucket, object_name)
+        with s3_fs.open(f's3://{bucket}/{object_name}', 'wb') as f:
+            f.write(file_content.read())
         return object_name
     except FileNotFoundError:
         print("The file was not found")
         return None
-    except NoCredentialsError:
-        print("Credentials not available")
+    except Exception as e:  # Catch broader exceptions that may arise from permissions or AWS issues
+        print(f"An error occurred: {e}")
         return None
 
+
 def process_image(file_content, s3_bucket, textract_client, object_name):
     s3_object_key = upload_file_to_s3(file_content, s3_bucket, object_name)
     if not s3_object_key: