Update app.py

app.py

@@ -19,7 +19,7 @@ def load_llm(max_tokens, prompt_template, temperature):
         temperature=temperature,
     )
 
-    llm_chain = LLMChain(llm=llm, prompt=prompt_template)
+    llm_chain = LLMChain(llm=llm, prompt=PromptTemplate.from_template(prompt_template))
     return llm_chain
 
 def get_src_original_url(query):
@@ -71,26 +71,26 @@ st.set_page_config(layout="wide")
 def main():
     st.title("GeniusWords: Unleash Your Imagination")
 
-    example_prompt_template = st.text_area("Please enter the example prompt template:")
+    example_prompt_template = 'You are a digital marketing and SEO expert and your task is to write an article on the given topic: {user_input}. The article must be under 800 words. The article should be lengthy.'
+    prompt_template = st.text_area("Please enter the prompt template:", value=example_prompt_template)
     max_tokens = st.number_input("Please enter the maximum tokens:", min_value=1, value=800)
     temperature = st.slider("Select the temperature:", min_value=0.1, max_value=1.0, value=0.7, step=0.1)
     user_input = st.text_input("Please enter the idea/topic for the article you want to generate!")
     image_input = st.text_input("Please enter the topic for the image you want to fetch!")
 
     if st.button("Generate Article"):
-        if len(user_input) > 0 and len(image_input) > 0 and len(example_prompt_template) > 0:
+        if len(user_input) > 0 and len(image_input) > 0 and len(prompt_template) > 0:
             col1, col2, col3 = st.columns([1, 2, 1])
             with col1:
                 st.subheader("Generated Content")
                 st.write("Topic of the article is: " + user_input)
                 st.write("Image of the article is: " + image_input)
 
-                prompt_template = example_prompt_template.format(user_input=user_input)
                 llm_call = load_llm(max_tokens=max_tokens, prompt_template=prompt_template, temperature=temperature)
                 result = llm_call(user_input)
                 if len(result) > 0:
                     st.info("Your article has been generated successfully!")
-                    st.write(result['text'])
+                    st.write(result)
                 else:
                     st.error("Your article couldn't be generated!")