Update app.py

app.py

@@ -3,13 +3,14 @@ import os
 import sys
 import json 
 import requests
+import random
 
 MODEL = "o1-preview"
 API_URL = os.getenv("API_URL")
 DISABLED = os.getenv("DISABLED") == 'True'
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+OPENAI_API_KEYS = os.getenv("OPENAI_API_KEYs").split(',')
 print (API_URL)
-print (OPENAI_API_KEY)
+print (OPENAI_API_KEYs)
 NUM_THREADS = int(os.getenv("NUM_THREADS"))
 
 print (NUM_THREADS)
@@ -47,7 +48,7 @@ def predict(inputs, top_p, temperature, chat_counter, chatbot, history, request:
 
     headers = {
         "Content-Type": "application/json",
-        "Authorization": f"Bearer {OPENAI_API_KEY}",
+        "Authorization": f"Bearer {random.choice(OPENAI_API_KEYS)}",
         "Headers": f"{request.kwargs['headers']}"
     }
 
@@ -121,7 +122,7 @@ def predict(inputs, top_p, temperature, chat_counter, chatbot, history, request:
 def reset_textbox():
     return gr.update(value='', interactive=False), gr.update(interactive=False)
 
-title = """<h1 align="center">GPT-4O: Research Preview (128K token limit, Short-Term Availability)</h1>"""
+title = """<h1 align="center">GPT-O1-Preview: Research Preview (Short-Term Availability)</h1>"""
 if DISABLED:
     title = """<h1 align="center" style="color:red">This app has reached OpenAI's usage limit. Please check back tomorrow.</h1>"""
 description = """Language models can be conditioned to act like dialogue agents through a conversational prompt that typically takes the form:
@@ -141,9 +142,7 @@ with gr.Blocks(css = """#col_container { margin-left: auto; margin-right: auto;}
                 #chatbot {height: 520px; overflow: auto;}""",
               theme=theme) as demo:
     gr.HTML(title)
-    #gr.HTML("""<h3 align="center">This app provides you full access to GPT-4O (128K token limit). You don't need any OPENAI API key.</h1>""")
-    #gr.HTML("""<h3 align="center" style="color: red;">If this app is too busy, consider trying our other GPT-4O app. Visit it below:<br/><a href="https://huggingface.co/spaces/yuntian-deng/ChatGPT4Turbo">https://huggingface.co/spaces/yuntian-deng/ChatGPT4Turbo</a></h3>""")
-    gr.HTML("""<h3 align="center" style="color: red;">If this app doesn't respond, consider trying our other GPT-4O app:<br/><a href="https://huggingface.co/spaces/yuntian-deng/ChatGPT4Turbo">https://huggingface.co/spaces/yuntian-deng/ChatGPT4Turbo</a></h3>""")
+    gr.HTML("""<h3 align="center" style="color: red;">If this app doesn't respond, consider trying our O1-mini app:<br/><a href="https://huggingface.co/spaces/yuntian-deng/ChatGPTo1mini">https://huggingface.co/spaces/yuntian-deng/ChatGPTo1mini</a></h3>""")
 
     #gr.HTML('''<center><a href="https://huggingface.co/spaces/ysharma/ChatGPT4?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>Duplicate the Space and run securely with your OpenAI API Key</center>''')
     with gr.Column(elem_id = "col_container", visible=False) as main_block: