Fix key handling, limit total messages to 20

.gitignore

@@ -0,0 +1 @@
+.history

app.py

@@ -14,9 +14,6 @@ def seed_submit(seed, history):
 
 def bot_update(history, key, alice_system, alice_temp, alice_model, bob_system, bob_temp, bob_model):  
 
-    if key:
-        os.environ['OPENAI_API_KEY'] = key
-
     while True:             
         if history[-1][1] is None:  # Bob's turn
             messages = [
@@ -27,6 +24,7 @@ def bot_update(history, key, alice_system, alice_temp, alice_model, bob_system,
                 if resp:
                     messages.append({"role": "assistant", "content": resp.replace(bob_prefix, '', 1)})
             resp = openai.ChatCompletion.create(
+                api_key=key,
                 model=bob_model, 
                 messages=messages,
                 temperature=bob_temp
@@ -43,13 +41,18 @@ def bot_update(history, key, alice_system, alice_temp, alice_model, bob_system,
                 if resp:
                     messages.append({"role": "user", "content": resp.replace(bob_prefix, '', 1)})
             resp = openai.ChatCompletion.create(
+                api_key=key,
                 model=alice_model, 
                 messages=messages,
                 temperature=alice_temp
             )            
             alice_response = resp['choices'][0]['message']['content']
             history.append([f'{alice_prefix}{alice_response}', None])
-        yield history
+
+        if len(messages) >= 10:
+            return history
+        else:
+            yield history
     
 
 with gr.Blocks() as demo: