Create app.py

app.py

@@ -0,0 +1,207 @@
+import os
+import gradio as gr
+from groq import Groq
+from huggingface_hub import InferenceClient
+from moviepy.editor import VideoFileClip
+import numpy as np
+from io import BytesIO
+
+# Lưu khóa API
+if 'GROQ_API_KEY' not in os.environ:
+    os.environ['GROQ_API_KEY'] = input('Nhập khóa API Groq của bạn: ')
+if 'HF_TOKEN' not in os.environ:
+    os.environ['HF_TOKEN'] = input('Nhập Hugging Face API Token của bạn: ')
+
+MAX_VIDEO_SIZE_MB = 35
+
+# Hàm gọi Groq API
+def call_groq_api(prompt, model_name="llama3-70b-8192", max_tokens=500):
+    client = Groq(api_key=os.environ.get('GROQ_API_KEY'))
+    response = client.chat.completions.create(
+        model=model_name,
+        messages=prompt,
+        max_tokens=max_tokens
+    )
+    return response.choices[0].message.content
+
+# Hàm trích xuất audio từ video
+def extract_audio_from_video(video_path):
+    try:
+        video = VideoFileClip(video_path)
+        audio_path = "temp_audio.mp3"
+        video.audio.write_audiofile(audio_path)
+        video.close()
+        return audio_path
+    except Exception as e:
+        raise Exception(f"Lỗi khi trích xuất audio: {str(e)}")
+
+# Hàm trích xuất thumbnails từ video
+def extract_thumbnails(video_path, num_frames=6):
+    try:
+        video = VideoFileClip(video_path)
+        duration = video.duration
+        step = duration / num_frames
+        thumbnails = []
+        for i in range(num_frames):
+            frame_time = i * step
+            frame = video.get_frame(frame_time)
+            thumbnails.append(frame)
+        video.close()
+        return thumbnails
+    except Exception as e:
+        raise Exception(f"Lỗi khi trích xuất thumbnails: {str(e)}")
+
+# Hàm gọi Hugging Face Inference API để chuyển audio thành văn bản
+def transcribe_audio(audio_path):
+    try:
+        client = InferenceClient(api_key=os.environ.get('HF_TOKEN'))
+        with open(audio_path, "rb") as audio_file:
+            transcription = client.automatic_speech_recognition(
+                audio_file,
+                model="openai/whisper-tiny"
+            )
+        return transcription['text']
+    except Exception as e:
+        raise Exception(f"Lỗi khi chuyển audio thành văn bản: {str(e)}")
+
+# Hàm tạo prompt cho Groq
+def create_prompt(social_media_type, transcription_text):
+    system_msg = (
+        "Bạn là chuyên gia trong việc tạo nội dung mạng xã hội và tạo bài đăng hiệu quả dựa trên nội dung người dùng. "
+        "Tuân thủ quy tắc và ràng buộc của nền tảng mạng xã hội."
+    )
+    user_msg = f"Nền tảng: {social_media_type}\nNội dung văn bản: {transcription_text}"
+    return [
+        {"role": "system", "content": system_msg},
+        {"role": "user", "content": user_msg}
+    ]
+
+# Hàm xử lý chính
+def process_and_generate_post(video_file, social_media_type, progress=gr.Progress()):
+    progress(0, desc="Đang khởi tạo...")
+    
+    if not video_file:
+        return "Vui lòng tải lên tệp video.", None, None, None
+    
+    try:
+        # Kiểm tra kích thước tệp video
+        video_size = os.path.getsize(video_file) / (1024 * 1024)  # Chuyển sang MB
+        if video_size > MAX_VIDEO_SIZE_MB:
+            return f"Tệp video lớn hơn {MAX_VIDEO_SIZE_MB} MB. Vui lòng tải lên tệp nhỏ hơn.", None, None, None
+        
+        progress(0.2, desc="Đang trích xuất audio...")
+        audio_path = extract_audio_from_video(video_file)
+        
+        progress(0.4, desc="Đang trích xuất thumbnails...")
+        thumbnails = extract_thumbnails(video_file)
+        
+        progress(0.6, desc="Đang chuyển audio thành văn bản...")
+        transcription_text = transcribe_audio(audio_path)
+        
+        progress(0.8, desc="Đang tạo bài đăng mạng xã hội...")
+        prompt = create_prompt(social_media_type, transcription_text)
+        social_media_post = call_groq_api(prompt)
+        
+        # Xóa tệp audio tạm
+        if os.path.exists(audio_path):
+            os.remove(audio_path)
+        
+        return social_media_post, thumbnails, transcription_text, audio_path
+    
+    except Exception as e:
+        return f"Đã xảy ra lỗi: {str(e)}", None, None, None
+
+# Giao diện Gradio
+def gradio_interface():
+    with gr.Blocks(theme=gr.themes.Base()) as demo:
+        gr.Markdown(
+            """
+            📹 Công cụ Tạo Bài Đăng Mạng Xã Hội từ Video
+            Biến video của bạn thành nội dung mạng xã hội hấp dẫn bằng AI
+            """
+        )
+
+        with gr.Row():
+            # Cột bên trái - Điều khiển đầu vào
+            with gr.Column():
+                video_input = gr.File(
+                    label="Tải lên Video",
+                    file_types=[".mp4", ".avi", ".mov"],
+                    height='400px'
+                )
+                
+                with gr.Group():
+                    gr.Markdown("### 🎯 Nền tảng Mục tiêu")
+                    social_media_type = gr.Radio(
+                        choices=[
+                            "X (Twitter)",
+                            "Facebook",
+                            "LinkedIn",
+                            "Instagram"
+                        ],
+                        value="X (Twitter)",
+                        label="Chọn nơi bạn muốn chia sẻ:",
+                        interactive=True
+                    )
+                
+                generate_btn = gr.Button(
+                    "🚀 Tạo Bài Đăng",
+                    scale=1,
+                    size="lg",
+                    variant="primary"
+                )
+
+            # Cột bên phải - Hiển thị kết quả
+            with gr.Column():
+                output = gr.Textbox(
+                    label="✨ Bài Đăng Đã Tạo",
+                    show_copy_button=True,
+                    lines=4
+                )
+                
+                gr.Markdown("### 🖼️ Hình thu nhỏ")
+                thumbnail = gr.Gallery(
+                    label="Chọn hình thu nhỏ ưa thích",
+                    show_download_button=True,
+                    show_fullscreen_button=True,
+                    height='200px',
+                    object_fit="contain"
+                )
+                
+                gr.Markdown("### 📝 Văn bản Đã Trích xuất")
+                df_output = gr.Textbox(
+                    label="Văn bản Đã Trích xuất",
+                    show_copy_button=True,
+                    lines=8
+                )
+                
+                gr.Markdown("### 🎵 Âm thanh Đã Trích xuất")
+                audio = gr.Audio(
+                    label="Âm thanh",
+                    show_download_button=True,
+                    type="filepath"
+                )
+
+        # Kết nối nút nhấn với hàm xử lý
+        generate_btn.click(
+            fn=process_and_generate_post,
+            inputs=[video_input, social_media_type],
+            outputs=[output, thumbnail, df_output, audio],
+            show_progress="full",
+            trigger_mode='once'
+        )
+
+    return demo
+
+# Khởi chạy giao diện Gradio
+if __name__ == "__main__":
+    demo = gradio_interface()
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        show_api=False,
+        show_error=False,
+        ssl_verify=True,
+        quiet=True
+    )