Update app.py

app.py

@@ -40,6 +40,7 @@ footer {
 
 LICENSE = '当前采用 ' + MODEL_NAME + ' 模型，请主动移除个人信息，注意隐私保护🔔'
 json_path = ""
+temp = ""
 
 local_data = threading.local()
 def init_data():
@@ -50,7 +51,7 @@ def init_data():
 def process_text(text_input, unit):
     init_data()
     client = Client(client_chat)
-    print(client.view_api())
+    # print(client.view_api())
     job = client.submit(
         query=str(text_input),
         history=None,
@@ -58,21 +59,21 @@ def process_text(text_input, unit):
         api_name="/model_chat"
     )
     response = job.result()
-    local_data.results = response[1][0][1]
+    temp = response[1][0][1]
+    local_data.results = temp
     return local_data.results
 
     
 
 def process_image(image_input, unit):
     init_data()
-    response = ""
     if image_input is not None:
         image = str(image_input)
         print(image)
         #with open(image_input, "rb") as f:
         #   base64_image = base64.b64encode(f.read()).decode("utf-8")
         client = Client(client_vl)
-        print(client.view_api())
+        # print(client.view_api())
         prompt = f" You are a experienced {unit} doctor AI assistant." + SYS_PROMPT + "Help me understand what is in this picture and analysis."
         
         res5 = client.predict(
@@ -99,29 +100,32 @@ def process_image(image_input, unit):
             response = job.result()
             with open(response, 'r') as f:
                 data = json.load(f)
-            print(data)
-            response = data[-1][1]
-        local_data.results = response     
-        threading.Thread(target=update).start()
-        return "正在分析....." 
+            temp = data[-1][1]
+
+        t = threading.Thread(target=update)
+        t.start()
+    return local_data.results = "正在分析....." 
         
 
 def fetch_result():
     init_data()
-    return local_data.results
+    if local_data.results:
+        local_data.results = temp
+        return local_data.results
     
 def reset_result():
     init_data()
     print(local_data.results)
-    local_data.results = 0
+    temp = ""
+    json_path = ""
+    local_data.results = ""
 
 def main(text_input="", image_input=None, unit=""):
-    init_data()
     reset_result()
     if text_input and image_input is None:
-        process_text(text_input, unit)
+        return process_text(text_input, unit)
     elif image_input is not None:
-        process_image(image_input, unit)
+        return process_image(image_input, unit)
 
 with gr.Blocks(css=css, title="家庭医生AI助手", theme="soft") as iface:
     with gr.Accordion(""):