|
import gradio as gr |
|
import random |
|
from gradio_client import Client |
|
|
|
def create_image(description): |
|
seed = 3343 + random.randint(0, 100000) |
|
image_client = Client("https://bytedance-hyper-sdxl-1step-t2i.hf.space/") |
|
result = image_client.predict( |
|
num_images=1, |
|
height=1024, |
|
width=1024, |
|
prompt=description, |
|
seed=seed, |
|
api_name="/process_image" |
|
) |
|
if result: |
|
image_url = result[0]["image"] |
|
return image_url |
|
else: |
|
return "Error: Could not retrieve image." |
|
|
|
def generate_content(description): |
|
content_client = Client("https://nick088-groq-api-chat.hf.space/") |
|
unchange_prompt = 'Chỉ viết nội dung và đưa kết quả nội dung, không cần diễn giải kể cả here is...' |
|
result = content_client.predict( |
|
message=description + unchange_prompt, |
|
request="llama3-70b-8192", |
|
param_3=0.5, |
|
param_4=4096, |
|
param_5=0.5, |
|
param_6=42, |
|
api_name="/chat" |
|
) |
|
return result |
|
|
|
def create_video(audio_bytes, image_url): |
|
video_client = Client("https://fffiloni-dreamtalk.hf.space/") |
|
audio_filepath = video_client.predict( |
|
audio_bytes, |
|
api_name="/load_audio" |
|
) |
|
result = video_client.predict( |
|
audio_filepath, |
|
image_url, |
|
"M030_front_neutral_level1_001.mat", |
|
api_name="/infer" |
|
) |
|
return result['video'] |
|
|
|
def main(): |
|
with gr.Blocks() as demo: |
|
with gr.Row(): |
|
gr.Markdown("# VMC 🇻🇳") |
|
gr.Image("https://huggingface.co/spaces/TDN-M/VMCBeta/resolve/main/demo/cover.jpeg") |
|
|
|
with gr.Row(): |
|
gr.Markdown(""" |
|
Dự án VMC (Virtual MC) là một ứng dụng đa nền tảng, cho phép người dùng tạo video với MC ảo từ hình ảnh và nội dung tuỳ chọn, sử dụng giọng nói tiếng Việt tự nhiên. |
|
## Sử dụng |
|
1. **Bước Tạo Ảnh:** Nhập mô tả về MC rồi nhấn nút "Tạo Ảnh". |
|
2. **Bước Xây Dựng Nội Dung:** Nhập mô tả nội dung và nhấn nút "Tạo Nội Dung". |
|
3. **Bước Phòng Thu:** Tải lên file âm thanh tham chiếu hoặc nhập URL âm thanh tham chiếu và nhấn nút "Tạo Âm Thanh". |
|
4. **Bước Tạo Video:** Nhấn nút "Tạo Video" để tạo video từ ảnh, nội dung, và âm thanh đã tạo. |
|
## Liên hệ |
|
Nếu bạn có bất kỳ câu hỏi nào, vui lòng liên hệ [dung.ngt1988@gmail.com](mailto:dung.ngt1988@gmail.com). |
|
""") |
|
|
|
with gr.Row(): |
|
gr.HTML(""" |
|
<style> |
|
.center-video { |
|
display: flex; |
|
justify-content: center; |
|
align-items: center; |
|
margin: 20px 0; |
|
} |
|
.center-video video { |
|
width: 50%; /* Giảm kích thước video xuống 50% */ |
|
} |
|
</style> |
|
""") |
|
gr.HTML(""" |
|
<div class="center-video"> |
|
<video controls> |
|
<source src="https://huggingface.co/spaces/TDN-M/VMCBeta/resolve/main/demo/demo2.mp4" type="video/mp4"> |
|
Trình duyệt của bạn không hỗ trợ video. |
|
</video> |
|
</div> |
|
""") |
|
|
|
with gr.Row(): |
|
description = gr.Textbox(label="Nhập mô tả cho ảnh bằng tiếng Anh") |
|
image_button = gr.Button("Tạo Ảnh") |
|
image_output = gr.Image() |
|
image_button.click(fn=create_image, inputs=description, outputs=image_output) |
|
|
|
with gr.Row(): |
|
content_description = gr.Textbox(label="Nhập mô tả nội dung") |
|
content_button = gr.Button("Tạo Nội Dung") |
|
content_output = gr.Textbox() |
|
content_button.click(fn=generate_content, inputs=content_description, outputs=content_output) |
|
|
|
with gr.Row(): |
|
gr.HTML('<iframe src="https://tdn-m-function-vmcbeta.hf.space" height="600"></iframe>') |
|
|
|
with gr.Row(): |
|
audio_output = gr.Audio(label="Tải lên âm thanh tham chiếu", type="filepath") |
|
video_button = gr.Button("Tạo Video") |
|
video_output = gr.Video() |
|
video_button.click(fn=create_video, inputs=[audio_output, image_output], outputs=video_output) |
|
|
|
demo.launch(share=True) |
|
if __name__ == "__main__": |
|
main() |
