| import os | |
| if not os.getenv("OPENAI_API_KEY"): | |
| raise ValueError("OPENAI_API_KEY must be set") | |
| from langchain.agents import initialize_agent | |
| from langchain.llms import OpenAI | |
| from gradio_tools import (StableDiffusionTool, ImageCaptioningTool, StableDiffusionPromptGeneratorTool, | |
| TextToVideoTool) | |
| from langchain.memory import ConversationBufferMemory | |
| llm = OpenAI(temperature=0) | |
| memory = ConversationBufferMemory(memory_key="chat_history") | |
| tools = [StableDiffusionTool().langchain, ImageCaptioningTool().langchain, | |
| StableDiffusionPromptGeneratorTool().langchain, TextToVideoTool().langchain] | |
| agent = initialize_agent(tools, llm, memory=memory, agent="conversational-react-description", verbose=True) | |
| output = agent.run(input=("Please create a photo of a dog riding a skateboard " | |
| "but improve my prompt prior to using an image generator." | |
| "Please caption the generated image and create a video for it using the improved prompt.")) |