s3 functionality

app.py

@@ -1,20 +1,26 @@
 import gradio
 import argparse
 import os
+import boto3
+from datetime import datetime
 
-from utils import generate
+from utils import generate, send_to_s3
 from models import get_tiny_llama, response_tiny_llama
 from constants import css, js_code, js_light
 
 MERA_table = None
 TINY_LLAMA = None
 
+S3_SESSION = None
+
 def giga_gen(content):
     res = generate(content,'auth_token.json')
+    send_to_s3(res, f'giga_{str(datetime.now()).replace(" ", "_")}.json', S3_SESSION)
     return res
 
 def tiny_gen(content):
     res = response_tiny_llama(TINY_LLAMA, content)
+    send_to_s3(res, f'tiny_{str(datetime.now()).replace(" ", "_")}.json', S3_SESSION)
     return res
 
 def tab_arena():
@@ -75,6 +81,18 @@ if __name__ == "__main__":
     # TYPES = ["number", "markdown", "number"]
 
     TINY_LLAMA =  get_tiny_llama()
+
+    try:
+        session = boto3.session.Session()
+        S3_SESSION = session.client(
+            service_name='s3',
+            endpoint_url=os.getenv('S3_ENDPOINT'),
+            aws_access_key_id=os.getenv('S3_ENDPOINT'),
+            aws_secret_access_key=os.getenv('S3_ENDPOINT'),
+        )
+    except:
+        print('Failed to start s3 session')
+
     demo = build_demo()
     demo.launch(share=args.share, height=3000, width="110%") # share=args.share
 

utils.py

@@ -2,6 +2,7 @@ import requests
 import json
 import os
 from datetime import datetime, timedelta
+import boto3
 
 
 def gen_auth_token(auth_file):
@@ -65,4 +66,8 @@ def generate(content='Привет!', auth_file=None):
 
     resp = get_text(content, auth_token['access_token'])
     
-    return resp["choices"][0]["message"]["content"]
+    return resp["choices"][0]["message"]["content"]
+
+def send_to_s3(data, name, session):
+    session.put_object(Bucket=os.getenv('S3_BUCKET'), Key=name, Body=json.dumps(data))
+