import gradio as gr import os import json import requests import openai import re #Streaming endpoint API_URL = "https://api.openai.com/v1/chat/completions" TWITTER_BEARER = os.getenv("TWITTER_BEARER") OPENAI_API_KEY = os.getenv("OPENAI_API_KEY") ##### stable diffusion stable_diffusion = gr.Blocks.load(name="spaces/runwayml/stable-diffusion-v1-5") def get_images(username): prompt_string = query(username) regexed_prompt_string = re.sub(r'@\w+', '', prompt_string) prompt_list = [element.split(". ")[1] for element in regexed_prompt_string.split(", ")] prompt_list_prepend = [f"an photo of {item}, photorealistic, coherent, high quality, rendered in unreal engine" for item in prompt_list] print(prompt_list_prepend) image_dirs = [] for prompt in prompt_list_prepend: image_dirs.append(stable_diffusion(prompt, fn_index=2)) sd_output = [] for image_dir in image_dirs: found_first_image = False for filename in os.listdir(image_dir): if filename.endswith('.json'): continue elif not found_first_image: sd_output.append(os.path.join(image_dir, filename)) found_first_image = True return sd_output, prompt_string with gr.Blocks(theme="gradio/soft") as demo: gr.Markdown("# 🎁 Starter Pack Generator 🎁") gr.Markdown("Enter a twitter username and this will return a starter pack for a person based on their latest tweets 🦜") gr.Markdown("Each generation might take up to a minute, please be patient, grab a green tea 🍵") gr.Markdown("This application uses Stable Diffusion space, so if you get `The queue is full!` please try again later, give the artist some break 🧑‍🎨") with gr.Row(): with gr.Column(): input_text = gr.Textbox(label="Input user ID", lines=4, elem_id="input-text") with gr.Row(): enter_username = gr.Button("Submit") with gr.Column(elem_id="generated-gallery"): sd_output = gr.Gallery().style(grid=3, height="auto") openai_output = gr.Textbox(lines=4) enter_username.click(get_images, inputs = [ input_text ], outputs = [sd_output, openai_output] ) ######### #### prompt creation def openai_create(prompt): openai.api_key=OPENAI_API_KEY response = openai.Completion.create( model="text-davinci-003", max_tokens = 100, prompt=prompt, stop="10.") return response.choices[0].text def query(username): bearer_token = TWITTER_BEARER headers = {"Authorization": "Bearer {}".format(bearer_token)} url = f"https://api.twitter.com/1.1/statuses/user_timeline.json?screen_name={username}&count={40}" try: response = requests.request("GET", url, headers = headers) print(response) twitter_stream = "" for tweet in response.json(): twitter_stream += tweet["text"] except Exception as e: print(e) result = ["Please make sure you are providing a correct, public twitter account"] if twitter_stream: twitter_stream = re.sub(r'@\w+', '', twitter_stream) prompt_text=f"""You will be given a long twitter stream belonging to a specific profile. I would like you to come up with a list of nine important concepts or items that user likes or is about them, and can be illustrated. Ignore any mentions to other people. Examples: Twitter stream: noughties Turkey was the coolest listening to this set for flow today 🏞️ this meme isn't brought to you by my hugging face persona, that one is below I often feel like what people really want vs what they say they want often mismatch. popular examples of this behavior these days, they want alexa to speak like generative models but they'd hate that if it happened. today we eat börek & kurabiye I feel like I will never ever love anything like I love mediterrenean 🌴🌊💙 I don't know if anyone else feels this way I'm ranked 10th on github trending today 😭 https://github.com/trending/developers paris is never cold, people are underdressed my all time favorite books are Meditations by Marcus Aurelius & Discourses by Epictetus, makes your skin thicker and you more resilient to challenges of life Concepts: 1. Turkish flag, 2. DJ set, 3. hugging face, 4. robot, 5. pastry, 6. mediterrenean, 7. github, 8. baguette bread, 9. meditations Twitter stream: Well im suck again, but imma stream some resident evil 4 if you wanna chill I thought wearing a turtle neck and gold chain was a meme Soon as I became old enough to wear "dad clothes" I dove right in. No hesitation. I've never been more comfortable in my life. All the video and audio goes from gaming pc over the network to the stream pc. Skips all that annoying audio wiring. No problems yet! The Silent Hill announcements are cool and all but the only things I've been able to rely on Konami to do in the past 10 years is aggressively claim music used in their videogames on YouTube, make pachinko machines, and cancel Silent Hill games. Every video I make THERE IS TOO MUCH TO DO IN ELDEN RING Overall my move to Australia has been pretty great. Went out for a Karaoke night with some friends but when it was my turn to sing. Concepts: 1. resident evil zombies, 2. a man wearing a turtle neck and gold chain, 3. dad clothes, 4. pc setup, 5. silent hill game, 6. youtube, 7. elden ring, 8. australia, 9. karaoke set Twitter stream: Meet Sonny! Our brand new coffee mascot! He’s brewing up some fun. This World Cup final is insane!!! I'm going to be on The Late Late Show in Ireland again this Friday! My next video will be my 5,000th upload CAT GAME 🐈 https://youtube.com/watch?v=OaE_UPIfiAo Happy Valentine’s Day ❤️ I hope you feel loved today, even if that love has to come from yourself. Treat yourself. Evelien surprised me 😊 Starting a Let’s Play channel with Tom Holland I normally never do reviews but I couldn’t NOT talk about Dune! https://youtu.be/yh9Evgt6ZBg @Spider-Man 2 Can’t wait!! Concepts: 1. coffee 2. football 3. Ireland 4. video 5. cat 6. love 7. gaming 8. Dune 9. spider-man Now, provide the list of 9 concepts for following twitter stream: Twitter stream: {twitter_stream} Concepts: """ result = openai_create(prompt=prompt_text) print(result) return result ####### demo.queue().launch(debug=True)