Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import requests | |
| def generate_video(prompt, duration, progress=gr.Progress()): | |
| """Generate video using akhaliq-Sora-2 API""" | |
| if not prompt.strip(): | |
| raise gr.Error("Please enter a text prompt") | |
| progress(0.1, desc="Starting video generation...") | |
| try: | |
| # API endpoint for akhaliq-Sora-2 | |
| api_url = "https://akhaliq-sora-2.hf.space/api/predict" | |
| # Prepare the payload | |
| payload = { | |
| "data": [ | |
| prompt, # text prompt | |
| duration, # duration in seconds | |
| 30, # fps (fixed) | |
| 512, # width (fixed) | |
| 512, # height (fixed) | |
| ] | |
| } | |
| progress(0.5, desc="Generating video...") | |
| # Make the API request | |
| response = requests.post(api_url, json=payload) | |
| if response.status_code == 200: | |
| progress(0.9, desc="Processing...") | |
| result = response.json() | |
| # Check if the generation is complete | |
| if "data" in result and len(result["data"]) > 0: | |
| video_url = result["data"][0] | |
| if video_url: | |
| progress(1.0, desc="Complete!") | |
| return video_url | |
| else: | |
| raise gr.Error("Video generation failed") | |
| else: | |
| raise gr.Error("Invalid response from API") | |
| else: | |
| raise gr.Error(f"API request failed with status code: {response.status_code}") | |
| except requests.exceptions.RequestException as e: | |
| raise gr.Error(f"Network error: {str(e)}") | |
| except Exception as e: | |
| raise gr.Error(f"Error generating video: {str(e)}") | |
| # Create the Gradio interface | |
| with gr.Blocks(title="Text-to-Video AI Generator") as demo: | |
| gr.Markdown("# 🎬 Text-to-Video AI Generator") | |
| gr.Markdown("Transform your text into AI-generated videos using Sora-2 model") | |
| gr.Markdown('<p><a href="https://huggingface.co/spaces/akhaliq/anycoder" target="_blank">Built with anycoder</a></p>') | |
| with gr.Row(): | |
| with gr.Column(): | |
| prompt_input = gr.Textbox( | |
| label="Video Description", | |
| placeholder="Describe the video you want to generate...", | |
| lines=3 | |
| ) | |
| duration_slider = gr.Slider( | |
| minimum=1, | |
| maximum=10, | |
| value=4, | |
| step=1, | |
| label="Video Duration (seconds)" | |
| ) | |
| generate_btn = gr.Button("Generate Video", variant="primary") | |
| with gr.Column(): | |
| video_output = gr.Video( | |
| label="Generated Video", | |
| height=400 | |
| ) | |
| gr.Examples( | |
| examples=[ | |
| ["A cat playing piano", 4], | |
| ["A robot dancing", 6], | |
| ["A sunset over the ocean", 3], | |
| ], | |
| inputs=[prompt_input, duration_slider] | |
| ) | |
| generate_btn.click( | |
| fn=generate_video, | |
| inputs=[prompt_input, duration_slider], | |
| outputs=[video_output], | |
| show_progress=True | |
| ) | |
| # Launch the app | |
| if __name__ == "__main__": | |
| demo.launch() |