Update app.py

app.py

@@ -10,7 +10,19 @@ def init_apis(openai_api_key, huggingface_api_key):
     # 这个函数用于初始化OpenAI和Hugging Face API
     openai.api_key = openai_api_key
     headers["Authorization"] = f"Bearer {huggingface_api_key}"
-    print("APIs initialized successfully.")
+    return "APIs initialized successfully."
+
+# 创建用于初始化API的Gradio接口
+init_interface = gr.Interface(
+    fn=init_apis,
+    inputs=[
+        gr.inputs.Textbox(label="OpenAI Key", type="password"),
+        gr.inputs.Textbox(label="HuggingFace Key", type="password"),
+    ],
+    outputs="text",
+    title="Initialize APIs",
+    description="Enter your OpenAI and Hugging Face API keys.",
+)
 
 def query(url):
     # 这个函数会向Hugging Face API发送图像URL，并获取图像中检测到的对象
@@ -21,10 +33,8 @@ def query(url):
     api_response = requests.request("POST", API_URL, headers=headers, data=data)
     return json.loads(api_response.content.decode("utf-8"))
 
-def process_query(openai_api_key, huggingface_api_key, user_query):
-    init_apis(openai_api_key, huggingface_api_key)
-
-    print("Processing the image object detection task...")
+def process_query(user_query):
+    # 这个函数处理图像对象检测任务
     function_descriptions = [
       {
           "name": "目标检测模型",
@@ -68,17 +78,16 @@ def process_query(openai_api_key, huggingface_api_key, user_query):
 
     return second_response['choices'][0]['message']['content']
 
-# Gradio界面
-iface = gr.Interface(
+# 创建用于处理查询的Gradio接口
+query_interface = gr.Interface(
     fn=process_query,
     inputs=[
-        gr.inputs.Textbox(label="OpenAI Key", type="password"),
-        gr.inputs.Textbox(label="HuggingFace Key", type="password"),
         gr.inputs.Textbox(label="Question")
     ],
     outputs="text",
-    title="Image Object Detection with Hugging Face and OpenAI",
-    description="Enter your OpenAI and Hugging Face API keys, and your question. The model will return the detected objects in the image.",
+    title="Process Query",
+    description="Enter your question. The model will return the detected objects in the image.",
 )
 
-iface.launch()
+init_interface.launch()
+query_interface.launch()