Update app.py

app.py

@@ -19,14 +19,31 @@ logger = logging.getLogger(__name__)
 def gradio_predict(image):
     try:
         # 使用临时文件目录保存图片
-        with tempfile.NamedTemporaryFile(suffix=".png", dir="/temp", delete=False) as temp_file:
-            temp_path = temp_file.name
-            logger.info(f"temp_path: {temp_path}")
+        # with tempfile.NamedTemporaryFile(suffix=".png", dir="/temp", delete=False) as temp_file:
+        #     temp_path = temp_file.name
+        #     logger.info(f"temp_path: {temp_path}")
+        #     image.save(temp_path)
+        #     logger.info(f"图片已保存至: {temp_path}")
+
+        # # 调用模型的 main 函数
+        if image is None:
+            logger.error("上传的图片为空")
+            return "错误: 请上传有效的图片文件"
+
+        # 使用 /home/user/app 保存临时文件
+        with tempfile.NamedTemporaryFile(suffix=".png", dir="/home/user/app", delete=False) as temp_file:
+            temp_path = os.path.abspath(temp_file.name)
+            logger.info(f"保存图片到: {temp_path}")
             image.save(temp_path)
-            logger.info(f"图片已保存至: {temp_path}")
 
-        # 调用模型的 main 函数
+        # 验证文件存在
+        if not os.path.exists(temp_path):
+            logger.error(f"图片文件未正确保存: {temp_path}")
+            return f"错误: 图片文件未正确保存 - {temp_path}"
 
+        # 检查路径内容（调试 't' 问题）
+        logger.info(f"传递给 main 的路径: {temp_path}, 类型: {type(temp_path)}")
+        
         smiles = predict_smiles(temp_path)
         logger.info(f"SMILES 输出: {smiles}")