Update app.py

app.py

@@ -104,6 +104,11 @@ from huggingface_hub import hf_hub_download
 #     logger.info(f'Received input: {input_text}')
 #     return "Output: " + input_text
 
+import sys
+
+# 设置无缓冲输出
+sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', buffering=1)
+sys.stderr = os.fdopen(sys.stderr.fileno(), 'w', buffering=1)
 
 # def get_render_cameras(batch_size=1, M=120, radius=2.5, elevation=10.0, is_flexicubes=False):
 #     """
@@ -590,15 +595,15 @@ def build_caption_anything_with_models(args, api_key="", captioner=None, sam_mod
 
 def validate_api_key(api_key):
     api_key = str(api_key).strip()
-    print(api_key)
+    print(api_key, flush=True)
     try:
         test_llm = ChatOpenAI(model_name="gpt-4o", temperature=0, openai_api_key=api_key)
-        print("test_llm")
+        print("test_llm", flush=True)
         response = test_llm([HumanMessage(content='Hello')])
-        print(response)
+        print(response, flush=True)
         return True
     except Exception as e:
-        print(f"API key validation failed: {e}")
+        print(f"API key validation failed: {e}", flush=True)
         return False
 
 
@@ -607,23 +612,23 @@ def init_openai_api_key(api_key=""):
     text_refiner = None
     visual_chatgpt = None
     if api_key and len(api_key) > 30:
-        print(api_key)
+        print(api_key, flush=True)
         if validate_api_key(api_key):
             try:
                 # text_refiner = build_text_refiner(args.text_refiner, args.device, args, api_key)
                 # assert len(text_refiner.llm('hi')) > 0 # test
                 text_refiner = None
-                print("text refiner")
+                print("text refiner", flush=True)
                 visual_chatgpt = ConversationBot(shared_chatbot_tools, api_key=api_key)
             except Exception as e:
-                print(f"Error initializing TextRefiner or ConversationBot: {e}")
+                print(f"Error initializing TextRefiner or ConversationBot: {e}", flush=True)
                 text_refiner = None
                 visual_chatgpt = None
         else:
-            print("Invalid API key.")
+            print("Invalid API key.", flush=True)
     else:
-        print("API key is too short.")
-    print(text_refiner)
+        print("API key is too short.", flush=True)
+    print(text_refiner, flush=True)
     openai_available = text_refiner is not None
     if visual_chatgpt: