Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import os
 import json
 import requests
+# from dotenv import load_dotenv
+# load_dotenv()
 
 API_URL = "https://api.openai.com/v1/chat/completions"
 
@@ -12,29 +14,22 @@ def get_system_txt():
     headers = {
         "Authorization": f"Bearer {hf_token}"
     }
-    response = requests.get(f"https://huggingface.co/spaces/clayp/reaction-sys/resolve/main/system.txt", headers=headers)
+    url = "https://huggingface.co/spaces/clayp/reaction-sys/resolve/main/system.txt"
+    response = requests.get(url, headers=headers)
+
     if response.status_code == 200:
         return response.text.strip()
     else:
         raise Exception("Failed to fetch system.txt from private Gradio space")
 
-system_msg = get_system_txt()
-
-def check_password(password):
-    if password == os.getenv("pw"):
-        return True
-    else:
-        return False
-
 def predict(inputs, chat_counter, chatbot=[], history=[]):
     headers = {
         "Content-Type": "application/json",
         "Authorization": f"Bearer {OPENAI_API_KEY}"
     }
     
-    # Read system message from system.txt
-    with open("system.txt", "r") as file:
-        system_msg = file.read().strip()
+    # Get system message using get_system_txt function
+    system_msg = get_system_txt()
     
     top_p = 1.0
     temperature = 0.0
@@ -90,17 +85,13 @@ title = """<h1 align="center">Response simulator</h1>"""
 with gr.Blocks() as demo:
     gr.HTML(title)
     with gr.Column(elem_id="col_container"):
-        password_input = gr.inputs.Password(label="Enter password to access the app")
-        if check_password(password_input):
-            chatbot = gr.Chatbot(label='GPT-4', elem_id="chatbot")
-            inputs = gr.Textbox(placeholder="", label="Type an input and press Enter")
-            state = gr.State([])
-            server_status_code = gr.Textbox(label="Status code from OpenAI server",)
-            chat_counter = gr.Number(value=0, visible=False, precision=0)
+        chatbot = gr.Chatbot(label='GPT-4', elem_id="chatbot")
+        inputs = gr.Textbox(placeholder="", label="Type an input and press Enter")
+        state = gr.State([])
+        server_status_code = gr.Textbox(label="Status code from OpenAI server",)
+        chat_counter = gr.Number(value=0, visible=False, precision=0)
 
-            inputs.submit(predict, [inputs, chat_counter, chatbot, state], [chatbot, state, chat_counter, server_status_code],)
-            inputs.submit(reset_textbox, [], [inputs])
-        else:
-            gr.Text("Incorrect password. Please try again.")
+    inputs.submit(predict, [inputs, chat_counter, chatbot, state], [chatbot, state, chat_counter, server_status_code], queue=True)
+    inputs.submit(reset_textbox, [], [inputs])
 
-demo.launch()
+demo.queue(max_size=5, concurrency_count=5).launch()