更新

app.py

@@ -17,17 +17,21 @@ def main():
     input_api_key = gr.inputs.Textbox(label="ChatGPT API Key", lines=1)
     input_api_base = gr.inputs.Textbox(label="ChatGPT API 地址（默认无地址）", lines=1)
     input_url = gr.inputs.Textbox(label="URL", lines=1)
+    def get_inputs(enabled):
+      if enabled:
+        return [input_api_key, input_api_base]
+      else:
+        return [input_url]
+    inputs = [input_checkbox] + get_inputs(input_checkbox.value)
+    outputs = "text"
+    update_interface = lambda enabled: gr.Interface(fn=my_inference_function, inputs=get_inputs(enabled), outputs=outputs)
+    interface = gr.Interface(fn=my_inference_function, inputs=inputs, outputs=outputs, examples=[["enabled", "api_key", "api_base", "url"]], title="ChatGPT")
+    interface.launch()
 
-    def my_inference_function(enabled, api_key, api_base, url):
+def my_inference_function(enabled, api_key, api_base, url):
         if enabled:
-            if api_key:
-                os.environ["OPENAI_API_KEY"] = api_key
-            else:
-                os.environ.pop("OPENAI_API_KEY", None)
-            if api_base:
-                os.environ['OPENAI_API_BASE'] = api_base
-            else:
-                os.environ.pop('OPENAI_API_BASE', None)
+            os.environ["OPENAI_API_KEY"] = api_key
+            os.environ['OPENAI_API_BASE'] = api_base
             llm = OpenAI(temperature=0.7, model_name="gpt-3.5-turbo", max_tokens=1024)
         else:
             llm = HuggingFaceHub(repo_id="declare-lab/flan-alpaca-large", model_kwargs={"temperature":0.1, "max_length":512})
@@ -42,18 +46,5 @@ def main():
 文章要点""")
         return response
 
-    def get_inputs(enabled):
-        if enabled:
-            return [input_api_key, input_api_base]
-        else:
-            return [input_url]
-
-    inputs = [input_checkbox] + get_inputs(input_checkbox.value)
-    outputs = "text"
-    update_interface = lambda enabled: gr.Interface(fn=my_inference_function, inputs=get_inputs(enabled), outputs=outputs)
-    interface = gr.Interface(fn=my_inference_function, inputs=inputs, outputs=outputs, examples=[["enabled", "api_key", "api_base", "url"]], title="ChatGPT")
-    interface.launch()
-
-
 if __name__ == '__main__':
     main()