New API

app.py

@@ -14,15 +14,17 @@ TINY_LLAMA = None
 
 S3_SESSION = None
 
-def giga_gen(content, history):
+def giga_gen(content, chat_history):
     res = generate(content,'auth_token.json')
+    chat_history.append((content, res))
     send_to_s3(res, f'protobench/giga_{str(datetime.now()).replace(" ", "_")}.json', S3_SESSION)
-    return res
+    return res, chat_history
 
-def tiny_gen(content, history):
+def tiny_gen(content, chat_history):
     res = response_tiny_llama(TINY_LLAMA, content)
+    chat_history.append((content, res))
     send_to_s3(res, f'protobench/tiny_{str(datetime.now()).replace(" ", "_")}.json', S3_SESSION)
-    return res
+    return res, chat_history
 
 
 def tab_arena():