ui

app.py

@@ -1,20 +1,31 @@
 import gradio as gr
+import os
 
-def greet(name, image):
-    # 这里我们只处理文本，忽略图片输入
-    return "Hello " + name + "!!"
+def process_image(image):
+    # 设置保存图片的路径，指向当前工程的 images 目录
+    save_path = os.path.join(os.getcwd(), "images")
+    # 确保目录存在
+    os.makedirs(save_path, exist_ok=True)
+    
+    # 为每个上传的图片生成唯一的文件名
+    from datetime import datetime
+    timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
+    image_filename = f"uploaded_{timestamp}.png"
+    image_path = os.path.join(save_path, image_filename)
+    
+    # 保存图片
+    image.save(image_path)
+    
+    return f"图片已保存至 {image_path}"  # 返回保存的位置信息
 
-# 定义输入：一个文本框和一个图片选择框
-inputs = [
-    gr.inputs.Textbox(label="Your Name"),
-    gr.inputs.Image(label="Your Image")
-]
+# 定义输入：一个图片选择框
+inputs = gr.inputs.Image(label="Upload Image", type="pil")
 
-# 定义输出：一个文本框
+# 定义输出：一个文本框，用来显示确认信息或处理结果
 outputs = gr.outputs.Textbox()
 
 # 创建 Interface 对象，设置 live=False 以添加提交按钮
-demo = gr.Interface(fn=greet, inputs=inputs, outputs=outputs, live=False)
+demo = gr.Interface(fn=process_image, inputs=inputs, outputs=outputs, live=False)
 
 # 启动界面
 demo.launch()