Update app.py

app.py

@@ -1,11 +1,33 @@
-import requests
-import numpy as np
+from PIL import Image, ImageDraw, ImageFont
+from dotenv import load_dotenv
+import matplotlib.pyplot as plt
 import gradio as gr
+import numpy as np
+import requests
+import boto3
+import uuid
+import os
 import io
-import matplotlib.pyplot as plt
-from PIL import Image, ImageDraw, ImageFont
+
+load_dotenv()
+AWS_ACCESS_KEY_ID = os.environ.get('AWS_ACCESS_KEY_ID')
+AWS_SECRET_ACCESS_KEY = os.environ.get('AWS_SECRET_ACCESS_KEY')
+s3 = boto3.client('s3',
+                  aws_access_key_id=AWS_ACCESS_KEY_ID,
+                  aws_secret_access_key=AWS_SECRET_ACCESS_KEY)
+
+   
+def upload2aws(img_array): 
+    image = Image.fromarray(img_array)
+    buffer = io.BytesIO()
+    image.save(buffer, format='JPEG')
+    buffer.seek(0)
+    unique_name = str(uuid.uuid4())
+    s3.put_object(Bucket='predict-packages', Key=f'images_webapp_counters/{unique_name}.jpg', Body=buffer)
+    return None
 
 def send2api(input_img, api_url):
+    upload2aws(input_img)
     buf = io.BytesIO()
     plt.imsave(buf, input_img, format='jpg')
     files = {'image': buf.getvalue()}