Upload app.py

app.py

@@ -119,21 +119,21 @@ def save_chat_history(session_id, history):
 def upload_to_gemini(path, mime_type=None):
     """上传文件到Gemini API"""
     file = genai.upload_file(path, mime_type=mime_type)
-    logger.debug(f"Uploaded file '{file.display_name}' as: {file.uri}")
+    debug.log_message(f"Uploaded file '{file.display_name}' as: {file.uri}")
     return file
 
 def wait_for_files_active(files):
     """等待文件处理完成"""
-    logger.debug("Waiting for file processing...")
+    debug.log_message("Waiting for file processing...")
     for name in (file.name for file in files):
         file = genai.get_file(name)
         while file.state.name == "PROCESSING":
-            logger.debug(".", end="", flush=True)
+            debug.log_message(".", end="", flush=True)
             time.sleep(10)
             file = genai.get_file(name)
         if file.state.name != "ACTIVE":
             raise Exception(f"File {file.name} failed to process")
-    logger.debug("...all files ready")
+    debug.log_message("...all files ready")
 
 @app.route('/upload', methods=['POST'])
 def get_upload_file():
@@ -286,7 +286,7 @@ def chat():
         files = []
         
         parts = userMessage.get("parts", [])
-        logger.debug(f"Parts: {parts}")
+        debug.log_message(f"Parts: {parts}")
         userTextMessage = parts[-1]
         file_urls = parts[:-1]
         if file_urls:
@@ -330,8 +330,8 @@ def chat():
             chat_history.append(model_message)
             save_chat_history(session_id, chat_history)
             debug.log_prompt(chat_history, preset_name)
-            logger.debug(f"User message parts: {parts}")
-            logger.debug(f"Chat history before sending: {chat_history}")
+            debug.log_message(f"User message parts: {parts}")
+            debug.log_message(f"Chat history before sending: {chat_history}")
             
         except Exception as e:
             logger.error(f"Error generating response: {str(e)}")