Spaces:
Running
Running
import gradio as gr | |
import cv2 | |
import numpy as np | |
from registry import registry | |
from filters import * | |
from components import create_filter_controls | |
def create_app(): | |
with gr.Blocks() as app: | |
gr.Markdown("# 📷 Real-Time Photo Filter App") | |
# Khởi tạo components | |
controls = create_filter_controls() | |
filter_names = list(registry.filters.keys()) | |
with gr.Row(): | |
with gr.Column(): | |
input_image = gr.Image(label="Input Image", type="numpy") | |
filter_select = gr.Dropdown( | |
label="Select Filter", | |
choices=filter_names, | |
value="Original" | |
) | |
# Thêm các control vào UI | |
control_components = [c for c in controls.values()] | |
# for component in control_components: # Removed render() calls | |
# component.render() | |
with gr.Column(): | |
output_image = gr.Image(label="Filtered Image") | |
# Xử lý cập nhật UI | |
def update_controls(filter_name): | |
updates = [] | |
for filter_name_ctrl, filter_ctrl_list in controls.items(): # Correctly unpack key-value pairs | |
visibility = filter_name_ctrl == filter_name | |
for component in filter_ctrl_list: # Iterate through the list of controls | |
updates.append(gr.update(visible=visibility)) # Update visibility for each control | |
return updates | |
# Xử lý ảnh real-time | |
def process(*args, **kwargs): | |
image = args[0] if len(args) > 0 else None | |
filter_name = args[1] if len(args) > 1 else "Original" # Default filter | |
if image is None: | |
return None | |
try: | |
image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR) | |
params = { | |
k.split('_', 1)[1]: v | |
for k, v in kwargs.items() | |
if k.startswith(filter_name) | |
} | |
processed = registry.filters[filter_name](image, **params) | |
if len(processed.shape) == 2: | |
processed = cv2.cvtColor(processed, cv2.COLOR_GRAY2RGB) | |
else: | |
processed = cv2.cvtColor(processed, cv2.COLOR_BGR2RGB) | |
return processed | |
except Exception as e: | |
print(f"Error processing image: {e}") | |
return image | |
# Kết nối sự kiện | |
filter_select.change( | |
update_controls, | |
inputs=filter_select, | |
outputs=list(controls.values()) | |
) | |
control_components_list = list(controls.values()) | |
input_components_process = [input_image, filter_select] + control_components_list | |
for component in [input_image, filter_select] + control_components_list: | |
component.change( | |
process, | |
inputs=input_components_process, | |
outputs=output_image, | |
show_progress=False | |
) | |
return app | |
if __name__ == "__main__": | |
app = create_app() | |
app.launch() |