| import gradio as gr | |
| import requests | |
| from PIL import Image | |
| import os | |
| import shutil | |
| # اطمینان از وجود پوشه temp | |
| if not os.path.exists('temp'): | |
| os.makedirs('temp') | |
| def tryon(image): | |
| # ذخیره تصویر دریافتی | |
| image.save("temp/upload.jpg") | |
| # ارسال تصویر به API پردازش | |
| files = {'image': open("temp/upload.jpg", "rb")} | |
| response = requests.post("http://localhost:7860/upload/", files=files) | |
| if response.ok: | |
| result_path = f"temp/{response.json()['file']}" | |
| return Image.open(result_path) | |
| else: | |
| return "خطا در پردازش" | |
| # رابط کاربری Gradio | |
| demo = gr.Interface(fn=tryon, inputs="image", outputs="image") | |
| demo.launch() |