add api base

app.py

@@ -324,10 +324,14 @@ def grade(file_obj, key, model, progress=gr.Progress()):
 
 
 # --- Validate key and model before running grading ---
-def validate_key_and_model(key: str, model: str):
-    openai.api_key = key.strip()  # strip leading/trailing spaces
+def validate_key_and_model(key: str, model: str, api_base: str = None):
+    openai.api_key = key.strip()
+    if api_base and api_base.strip():
+        openai.api_base = api_base.strip()   # 用户自定义 api_base
+    else:
+        openai.api_base = "https://api.openai.com/v1"  # 默认官方 OpenAI
+
     try:
-        # This call is fast and checks both key validity and model availability
         openai.Model.retrieve(model)
         return True, "OK"
     except openai.error.AuthenticationError:
@@ -347,10 +351,9 @@ def validate_key_and_model(key: str, model: str):
         return False, f"Unexpected error: {e}"
 
 # --- Wrapper for the grading function ---
-def run_grade(file_obj, key, model, progress=gr.Progress(track_tqdm=True)):
-    ok, msg = validate_key_and_model(key, model)
+def run_grade(file_obj, key, model, api_base, progress=gr.Progress(track_tqdm=True)):
+    ok, msg = validate_key_and_model(key, model, api_base)
     if not ok:
-        # This will be visible to the user in the Gradio UI
         raise gr.Error(msg)
     return grade(file_obj, key, model, progress=progress)
 
@@ -376,6 +379,10 @@ The grading results will be downloaded as a zip file.
 with gr.Blocks() as demo:
     gr.Markdown(markdown)
     key = gr.Textbox(label=f"Enter your OpenAI API Key (this space will not save your API Key). [Pay Attention]: this evaluaiton may cost several dollars, please notice your OpenAI API Key balance.", type="password")
+    api_base = gr.Textbox(
+        label="Enter your OpenAI API Base (leave empty to use official OpenAI)",
+        value=""
+    )
     model = gr.Dropdown(
         choices=["gpt-4-0613", "gpt-4.1", "gpt-4-turbo"],
         value="gpt-4.1",
@@ -385,7 +392,7 @@ with gr.Blocks() as demo:
         inp = gr.File(file_types=[".json"])
         out = gr.File(file_types=[".zip"])
     btn = gr.Button("Start grading", variant="primary")
-    btn.click(fn=run_grade, inputs=[inp, key, model], outputs=out)
+    btn.click(fn=run_grade, inputs=[inp, key, model, api_base], outputs=out)
 
 if __name__ == "__main__":
     demo.queue(max_size=8).launch()