| import requests |
| import base64 |
| import json |
|
|
| |
| ENDPOINT_URL = "https://your-endpoint-url.huggingface.co" |
| API_TOKEN = "your-hf-token-here" |
|
|
|
|
| def encode_file_to_base64(file_path): |
| """Convert file to base64 string""" |
| with open(file_path, "rb") as f: |
| return base64.b64encode(f.read()).decode() |
|
|
|
|
| |
| def test_endpoint(): |
| headers = { |
| "Authorization": f"Bearer {API_TOKEN}", |
| "Content-Type": "application/json", |
| } |
|
|
| |
| image_base64 = encode_file_to_base64("path/to/your/image.png") |
| audio_spk1_base64 = encode_file_to_base64("path/to/speaker1.wav") |
| audio_spk2_base64 = encode_file_to_base64("path/to/speaker2.wav") |
|
|
| |
| payload = { |
| "inputs": { |
| "prompt": "A person speaking about technology", |
| "image": image_base64, |
| "audio_spk1": audio_spk1_base64, |
| "audio_spk2": audio_spk2_base64, |
| }, |
| "parameters": {"sample_steps": 12}, |
| } |
|
|
| print("Sending request to endpoint...") |
| print(f"Prompt: {payload['inputs']['prompt']}") |
| print(f"Sample steps: {payload['parameters']['sample_steps']}") |
|
|
| |
| response = requests.post(ENDPOINT_URL, headers=headers, json=payload) |
|
|
| if response.status_code == 200: |
| result = response.json() |
| if "generated_video" in result: |
| |
| video_data = base64.b64decode(result["generated_video"]) |
| with open("output_video.mp4", "wb") as f: |
| f.write(video_data) |
| print("✅ Success! Video saved as output_video.mp4") |
| else: |
| print("❌ Error in response:", result.get("error", "Unknown error")) |
| else: |
| print(f"❌ Request failed with status {response.status_code}") |
| print(response.text) |
|
|
|
|
| if __name__ == "__main__": |
| test_endpoint() |
|
|
|
|
| |
| |
| |
|
|
| |
| single_speaker_request = { |
| "inputs": { |
| "prompt": "A person talking about the weather", |
| "image": "base64_encoded_image_data_here", |
| "audio_spk1": "base64_encoded_audio_data_here", |
| }, |
| "parameters": {"sample_steps": 12}, |
| } |
|
|
| |
| two_speakers_request = { |
| "inputs": { |
| "prompt": "Two people having a conversation", |
| "image": "base64_encoded_image_data_here", |
| "audio_spk1": "base64_encoded_audio1_data_here", |
| "audio_spk2": "base64_encoded_audio2_data_here", |
| }, |
| "parameters": {"sample_steps": 15}, |
| } |
|
|
| |
| expected_response = { |
| "generated_video": "base64_encoded_video_data", |
| "content_type": "video/mp4", |
| } |
|
|
| |
| error_response = {"error": "Error message here"} |
|
|
| print("\n" + "=" * 50) |
| print("SAMPLE REQUEST FORMATS:") |
| print("=" * 50) |
| print("\n1. Single Speaker Request:") |
| print(json.dumps(single_speaker_request, indent=2)) |
| print("\n2. Two Speakers Request:") |
| print(json.dumps(two_speakers_request, indent=2)) |
| print("\n3. Expected Success Response:") |
| print(json.dumps(expected_response, indent=2)) |
| print("\n4. Expected Error Response:") |
| print(json.dumps(error_response, indent=2)) |
|
|
