Create caption_chain.py

components/caption_chain.py

@@ -0,0 +1,11 @@
+from langchain import PromptTemplate, LLMChain
+
+def chain(llm):
+    template = """Make 5 different advertisement captions about this product. 
+    {product}
+    """
+    
+    prompt = PromptTemplate(template=template, input_variables=["product"])
+    
+    llm_chain = LLMChain(prompt=prompt, llm=llm)
+    return llm_chain