error

app.py

@@ -1,10 +1,8 @@
-import gradio as gr 
-from typing import List
+import gradio as gr
 from utils import get_base_answer, get_nudging_answer
 from constant import js_code_label, custom_css, HEADER_MD, BASE_MODELS, NUDGING_MODELS
 import datetime
 import logging
-import urllib.request
 # add logging info to console 
 logging.basicConfig(level=logging.INFO)
 
@@ -29,8 +27,8 @@ def respond_base(
     host_addr = request.client.host
     if host_addr not in addr_limit_counter:
         addr_limit_counter[host_addr] = 0
-    if addr_limit_counter[host_addr] > 1:
-        return gr.Error("You have reached the limit of 50 requests for today. Please use your own API key.")
+    if addr_limit_counter[host_addr] > 0:
+        raise gr.Error("You have reached the limit of 50 requests for today. Please use your own API key.", duration=10)
 
     base_answer = get_base_answer(base_model=base_model, system_prompt=system_prompt, question=message, max_tokens=max_tokens)
     addr_limit_counter[host_addr] += 1
@@ -56,8 +54,8 @@ def respond_nudging(
     host_addr = request.client.host
     if host_addr not in addr_limit_counter:
         addr_limit_counter[host_addr] = 0
-    if addr_limit_counter[host_addr] > 1:
-        return gr.Error("You have reached the limit of 50 requests for today. Please use your own API key.")
+    if addr_limit_counter[host_addr] > 0:
+        raise gr.Error("You have reached the limit of 50 requests for today. Please use your own API key.", duration=10)
     all_info = get_nudging_answer(base_model=base_model, nudging_model=nudging_model, system_prompt=system_prompt, question=message, max_token_total=max_tokens, top_prob_thres=nudging_thres)
     all_completions = all_info["all_completions"]
     nudging_words = all_info["all_nudging_words"]