Update app.py

app.py

@@ -19,9 +19,8 @@ def visualize_molecule(smiles):
     except:
         return None
 
-def gradio_predict(image):
+def predict(image):
     try:
-        # 检查图片输入
         if image is None:
             logger.error("上传的图片为空")
             return "错误: 请上传有效的图片文件"
@@ -32,22 +31,19 @@ def gradio_predict(image):
             logger.error(f"工作目录不存在或不是目录: {work_dir}")
             return f"错误: 工作目录无效 - {work_dir}"
 
-        # 使用 /home/user/app 保存临时文件
         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}, 类型: {type(temp_path)}")
             image.save(temp_path)
 
-        # 验证文件存在
         if not os.path.exists(temp_path):
             logger.error(f"图片文件未正确保存: {temp_path}")
             return f"错误: 图片文件未正确保存 - {temp_path}"
 
-        # 调用模型的 main 函数
         logger.info(f"传递给 main 的路径: {temp_path}")
         smiles = predict_smiles([temp_path])
         mol_image = visualize_molecule(smiles)
-        # 清理临时文件
+
         try:
             os.remove(temp_path)
             logger.info(f"已删除临时文件: {temp_path}")
@@ -74,15 +70,17 @@ def gradio_predict(image):
 
 # 定义 Gradio 界面
 iface = gr.Interface(
-    fn=gradio_predict,
-    inputs=gr.Image(type="pil"),
-    # outputs=gr.Textbox(),
+    fn=predict,
+    inputs=gr.Image(label="Upload molecular image", type="pil"),
     outputs=[
-        gr.Image(label="Prediction"),
+        gr.Image(label="Mol Image"),
         gr.Text(label="SMILES"),
     ],
-    title="化学结构转 SMILES",
-    description="上传化学结构图片以获取其 SMILES 字符串"
+    title="Optical Chemical Structure Recognition",
+    description="Convert a molecular image into SMILES.<br> ",
+    examples=[
+        ["example.png"]
+    ]
 )
 
 if __name__ == "__main__":