Update app.py

app.py

@@ -10,24 +10,56 @@ sys.path.append(os.path.dirname(os.path.abspath(__file__)))
 from ONNX0630 import main as predict_smiles
 from PIL import Image
 import io
+import tempfile
+import logging
 
+# 设置日志
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 # Gradio interface
+# def gradio_predict(image):
+#     try:
+#         # Save the uploaded image
+#         temp_path = "temp_image.png"
+#         image.save(temp_path)
+        
+#         # Call the model function
+#         smiles = predict_smiles(temp_path)
+        
+#         # Clean up
+#         os.remove(temp_path)
+        
+#         return smiles
+#     except Exception as e:
+#         return f"Error: {str(e)}"
+    
 def gradio_predict(image):
     try:
-        # Save the uploaded image
-        temp_path = "temp_image.png"
-        image.save(temp_path)
-        
-        # Call the model function
+        # 使用临时文件目录保存图片
+        with tempfile.NamedTemporaryFile(suffix=".png", dir="/app", delete=False) as temp_file:
+            temp_path = temp_file.name
+            image.save(temp_path)
+            logger.info(f"图片已保存至: {temp_path}")
+
+        # 调用模型的 main 函数
         smiles = predict_smiles(temp_path)
-        
-        # Clean up
+        logger.info(f"SMILES 输出: {smiles}")
+
+        # 清理临时文件
         os.remove(temp_path)
-        
+        logger.info(f"已删除临时文件: {temp_path}")
+
         return smiles
+    except FileNotFoundError as e:
+        logger.error(f"文件路径错误: {e}")
+        return f"错误: 文件路径问题 - {str(e)}"
+    except PermissionError as e:
+        logger.error(f"权限错误: {e}")
+        return f"错误: 文件权限问题 - {str(e)}"
     except Exception as e:
-        return f"Error: {str(e)}"
+        logger.error(f"处理错误: {e}")
+        return f"错误: {str(e)}"
 
 # Define Gradio interface
 iface = gr.Interface(