File size: 1,535 Bytes
ae67512 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | import logging
import json
from main import ViralClipExtractor, extract_video_id
logging.basicConfig(level=logging.INFO)
def test():
url = "https://youtu.be/uVkFrqugXFQ"
nvidia_key = "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"
print("=== FASTAPI LOCAL TEST ===")
print(f"URL: {url}")
extractor = ViralClipExtractor()
# 1. Video Info
print("\n1. Fetching Video Info...")
video_info = extractor.extract_video_info(url)
print(f" Title: {video_info.get('title')}")
print(f" Duration: {video_info.get('duration')} seconds")
# 2. Transcript
print("\n2. Fetching Transcript...")
transcript = extractor.fetch_full_transcript(url)
print(f" Transcript Segments: {len(transcript)}")
if transcript:
print(f" Sample: {transcript[0]}")
else:
print(" [FAIL] No transcript found!")
return
# 3. AI Analysis
print("\n3. Running Nvidia AI Analysis...")
clips = extractor.analyze_with_nvidia(transcript, nvidia_key)
print(f" Clips Found: {len(clips)}")
# 4. Final Response
response = {
"success": True,
"video_id": extract_video_id(url),
"video_title": video_info.get('title'),
"video_duration": video_info.get('duration'),
"mode": "nvidia",
"clips": clips,
"clips_count": len(clips)
}
print("\n=== FINAL RESPONSE ===")
print(json.dumps(response, indent=2))
if __name__ == "__main__":
test()
|