Create utils.py

utils.py

@@ -0,0 +1,29 @@
+from langchain.prompts import PromptTemplate
+from langchain.chains import LLMChain
+from langchain.tools import DuckDuckGoSearchRun
+from langchain_google_genai import GoogleGenerativeAI
+
+def generate_script(prompt,video_length,creativity,api_key):
+
+    title_template = PromptTemplate(
+        input_variable = ['subject'],
+        template = 'Please come up with a title for a youtube video on the {subject}.'
+    )
+
+    script_template = PromptTemplate(
+        input_variable = ['title','DuckDuckGo_Search','duration'],
+        template = "Create a script for a youtube video based on this title for me. TITLE: {title} of duration: {duration} minutes using this search data {DuckDuckGo_Search}"
+    )
+
+    llm = GoogleGenerativeAI(temperature=creativity,google_api_key=api_key,model = "gemini-pro")
+
+    title_chain = LLMChain(llm=llm, prompt=title_template,verbose = True)
+    script_chain = LLMChain(llm=llm, prompt=script_template,verbose = True)
+
+    search = DuckDuckGoSearchRun()
+
+    title = title_chain.run(prompt)
+    search_result = search.run(prompt)
+    script = script_chain.run(title=title,DuckDuckGo_Search=search_result,duration = video_length)
+
+    return search_result,title,script