Update app.py

app.py

@@ -46,8 +46,8 @@ def process_queue():
                 label, score = classify_text(request['text'])
                 request['response'].append((label, score))
             else:
-                # Put it back in the queue
-                non_authenticated_queue.put(request)
+                # Put it back in the queue if there are authenticated requests
+                authenticated_queue.put(request)
         time.sleep(0.1)  # Adjust the sleep time as necessary
 
 def queue_request(text, api_key):
@@ -68,6 +68,9 @@ def queue_request(text, api_key):
     
     return response[0]
 
+def get_queue_sizes():
+    return len(authenticated_queue.queue), len(non_authenticated_queue.queue)
+
 # Start the queue processing thread
 threading.Thread(target=process_queue, daemon=True).start()
 
@@ -83,5 +86,15 @@ interface = gr.Interface(
     description="A simple app to classify text using the roberta-mixed-detector model with priority queuing."
 )
 
+queue_interface = gr.Interface(
+    fn=get_queue_sizes,
+    inputs=[],
+    outputs=[gr.Number(label="Authenticated Queue Size"), gr.Number(label="Non-Authenticated Queue Size")],
+    live=True
+)
+
+# Combine the interfaces
+combined_interface = gr.TabbedInterface([interface, queue_interface], ["Classify Text", "Queue Sizes"])
+
 # Launch the app
-interface.launch()
+combined_interface.launch()