Hritz1999 commited on
Commit
09b945c
Β·
1 Parent(s): 707b487

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +57 -0
app.py ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Bring in deps
2
+ import os
3
+ #from apikey import apikey
4
+
5
+ import streamlit as st
6
+ from langchain.llms import OpenAI
7
+ from langchain.prompts import PromptTemplate
8
+ from langchain.chains import LLMChain, SequentialChain
9
+ from langchain.memory import ConversationBufferMemory
10
+ from langchain.utilities import WikipediaAPIWrapper
11
+ OpenAI.organization = "org-y2x79N0uXGd9k3wnv1SBwfll"
12
+ os.environ['OPENAI_API_KEY'] = apikey
13
+
14
+ # App framework
15
+ st.title('πŸ¦œπŸ”— YouTube GPT Creator')
16
+ prompt = st.text_input('Plug in your prompt here')
17
+
18
+ # Prompt templates
19
+ title_template = PromptTemplate(
20
+ input_variables = ['topic'],
21
+ template='write me a youtube video title about {topic}'
22
+ )
23
+
24
+ script_template = PromptTemplate(
25
+ input_variables = ['title', 'wikipedia_research'],
26
+ template='write me a youtube video script based on this title TITLE: {title} while leveraging this wikipedia reserch:{wikipedia_research} '
27
+ )
28
+
29
+ # Memory
30
+ title_memory = ConversationBufferMemory(input_key='topic', memory_key='chat_history')
31
+ script_memory = ConversationBufferMemory(input_key='title', memory_key='chat_history')
32
+
33
+
34
+ # Llms
35
+ llm = OpenAI(temperature=0.9)
36
+ title_chain = LLMChain(llm=llm, prompt=title_template, verbose=True, output_key='title', memory=title_memory)
37
+ script_chain = LLMChain(llm=llm, prompt=script_template, verbose=True, output_key='script', memory=script_memory)
38
+
39
+ wiki = WikipediaAPIWrapper()
40
+
41
+ # Show stuff to the screen if there's a prompt
42
+ if prompt:
43
+ title = title_chain.run(prompt)
44
+ wiki_research = wiki.run(prompt)
45
+ script = script_chain.run(title=title, wikipedia_research=wiki_research)
46
+
47
+ st.write(title)
48
+ st.write(script)
49
+
50
+ with st.expander('Title History'):
51
+ st.info(title_memory.buffer)
52
+
53
+ with st.expander('Script History'):
54
+ st.info(script_memory.buffer)
55
+
56
+ with st.expander('Wikipedia Research'):
57
+ st.info(wiki_research)