j dZddlZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z m Z dedefdZddedzdefd Zdd ed edzdeeeffd Zed k(rej*dZej/ddej/ddej1Zeeej4ej6\ZZeej>eede ed jCddyy)zLCreate an original, fast-paced story suited to a 45-65 second YouTube Short.N)datetimetimezone)OpenAI)OPENAI_API_KEYOPENAI_BASE_URL OPENAI_MODEL STORIES_DIR STORY_THEMESthemereturncd}d|dgd|ddS)z>Works without an API key so the video renderer remains usable.aMy coworker kept stealing my lunch, and everyone told me I was imagining it. For three months, my meals disappeared from the fridge every Tuesday. So I put a tiny Bluetooth tracker inside a sealed container of leftovers. At noon, the signal led straight to our manager's office. He had been taking food because he thought nobody would confront him. I showed HR the location history, and suddenly he had a lot to explain. The next day, he brought in a whole tray of sandwiches for the team. Nobody touched them.z!My Manager Kept Stealing My LunchzPA workplace mystery with an unexpected ending. #storytime #redditstories #shorts)z#shortsz #storytimez#redditstoriesz#karmazoriginal-fallback)titlestory descriptionhashtagsr source)r texts !/root/workspace/ytshorts/story.pyfallback_storyr s*  5iI%  c |xstjt}ts t |St tt }d|d} |jjjtd|dgdddi }tj|jd jjt!fd d Ds t#d |d<dd<S#t$$r$}t'd|dt |cYd}~Sd}~wwxYw)N)api_keybase_urlzAWrite an original first-person story for a YouTube Short. Theme: a~ Rules: - 125 to 150 spoken English words, target 50 to 60 seconds. - Begin with a compelling hook in the first sentence. - Use natural spoken language, short sentences, and escalating stakes. - End with a satisfying, surprising, or ironic payoff. - Entirely fictional and original. Do not reference Reddit, real people, brands, or existing posts. - No graphic violence, sexual content, self-harm, hate, or slurs. Return ONLY valid JSON matching this shape: {"title":"under 65 characters", "story":"the narration text", "description":"under 150 characters, includes 2-4 relevant hashtags", "hashtags":["#shorts", "#storytime", "#... "]}user)rolecontentg?type json_object)modelmessages temperatureresponse_formatrc3@K|]}j|ywN)get).0krs r z!generate_story..=sWA599Q<Ws)rrrrz&Model response omitted required fieldsr z original-airzStory API failed (z"); using the local fallback story.)randomchoicer rrrrchat completionscreaterjsonloadschoicesmessagerall ValueError Exceptionprint)r clientpromptresponseexcrs @rgenerate_storyr;"s   0V]]<0E e$$ N_ EF w u xF%;;**11%&9:#]3 2   8++A.66>>?W)VWWEF Fg'h % "3%'IJKe$$%s BC D )DD D rstemcNtjtjj d}|xs|}tjtjj |d<t |dz }|jtj|dddz||fS)Nz %Y%m%d_%H%M%S created_atz.jsonT)indent ensure_ascii ) rnowrutcstrftime isoformatr write_textr/dumps)rr< timestamppaths r save_storyrKGs X\\*33ODI  9D",,x||4>>@E, D6 'DOODJJuQTBTIJ $;r__main__z Generate an original Short story)rz--themez Story theme)helpz --output-stemzOutput filename stemr)r<rwordsr?)r@r%)"__doc__argparser/r*rropenairconfigrrrr r strdictrr;tuplerK__name__ArgumentParserparser add_argument parse_argsargsr output_stemrr<r6rHlensplitrrrr_sR '[[#$,"%#*"%"%Jd#*dCi8H z $X $ $1S TF  6 .DE    D^DJJ79I9IJKE4 *$**dU7^c%PW.J^J^J`Fabkl mn r