Update app.py

app.py

@@ -1,39 +1,33 @@
-from fastapi import FastAPI, File, UploadFile
-from fastapi.responses import JSONResponse
-import gradio as gr
-import os
-
 import sys
 sys.path.append(os.path.dirname(os.path.abspath(__file__)))
-
+import os
+import gradio as gr
 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__)
-   
+
+
 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}")
-        #     image.save(temp_path)
-        #     logger.info(f"图片已保存至: {temp_path}")
-
-        # # 调用模型的 main 函数
+        # 检查图片输入
         if image is None:
             logger.error("上传的图片为空")
             return "错误: 请上传有效的图片文件"
 
+        # 验证工作目录
+        work_dir = "/home/user/app"
+        if not os.path.isdir(work_dir):
+            logger.error(f"工作目录不存在或不是目录: {work_dir}")
+            return f"错误: 工作目录无效 - {work_dir}"
+
         # 使用 /home/user/app 保存临时文件
-        with tempfile.NamedTemporaryFile(suffix=".png", dir="/home/user/app", delete=False) as temp_file:
+        with tempfile.NamedTemporaryFile(suffix=".png", dir=work_dir, delete=False) as temp_file:
             temp_path = os.path.abspath(temp_file.name)
-            logger.info(f"保存图片到: {temp_path}")
+            logger.info(f"保存图片到: {temp_path}, 类型: {type(temp_path)}")
             image.save(temp_path)
 
         # 验证文件存在
@@ -41,20 +35,24 @@ def gradio_predict(image):
             logger.error(f"图片文件未正确保存: {temp_path}")
             return f"错误: 图片文件未正确保存 - {temp_path}"
 
-        # 检查路径内容（调试 't' 问题）
-        logger.info(f"传递给 main 的路径: {temp_path}, 类型: {type(temp_path)}")
-        
+        # 调用模型的 main 函数
+        logger.info(f"传递给 main 的路径: {temp_path}")
         smiles = predict_smiles(temp_path)
-        logger.info(f"SMILES 输出: {smiles}")
 
         # 清理临时文件
-        os.remove(temp_path)
-        logger.info(f"已删除临时文件: {temp_path}")
+        try:
+            os.remove(temp_path)
+            logger.info(f"已删除临时文件: {temp_path}")
+        except Exception as e:
+            logger.warning(f"删除临时文件失败: {e}")
 
         return smiles
     except FileNotFoundError as e:
         logger.error(f"文件路径错误: {e}")
         return f"错误: 文件路径问题 - {str(e)}"
+    except IsADirectoryError as e:
+        logger.error(f"路径是目录: {e}")
+        return f"错误: 路径是目录 - {str(e)}"
     except PermissionError as e:
         logger.error(f"权限错误: {e}")
         return f"错误: 文件权限问题 - {str(e)}"
@@ -62,19 +60,14 @@ def gradio_predict(image):
         logger.error(f"处理错误: {e}")
         return f"错误: {str(e)}"
 
-# Define Gradio interface
+# 定义 Gradio 界面
 iface = gr.Interface(
     fn=gradio_predict,
     inputs=gr.Image(type="pil"),
     outputs=gr.Textbox(),
-    title="Chemical Structure to SMILES Converter",
-    description="Upload an image of a chemical structure to get its SMILES string."
+    title="化学结构转 SMILES",
+    description="上传化学结构图片以获取其 SMILES 字符串"
 )
 
-# Launch Gradio with FastAPI
-# app = gr.mount_gradio_app(app, iface, path="/")
-
 if __name__ == "__main__":
-    # import uvicorn
-    # uvicorn.run(app, host="0.0.0.0", port=7860)
-    iface.launch(server_name="0.0.0.0", server_port=7860)
+    iface.launch(server_name="0.0.0.0", server_port=7860, share=False)