prompt changes

README.md

@@ -9,8 +9,12 @@ app_file: app.py
 pinned: false
 ---
 
-# Blogger
+# The Blogger 📝
 
-LLAMA2 - Article Generation
+[The Blogger 📝 - Playground](https://huggingface.co/spaces/thivav/llama2-blogger)
 
-[The TheBloke/Llama-2-7B-Chat-GGML llama2 model used](https://huggingface.co/TheBloke/Llama-2-7B-Chat-GGML)
+This application generates an article based on your query and gener.
+
+- For **The Blogger** following technologies were used:
+    - **llama2** with **langchain** used to generate the blog.
+        - [The TheBloke/Llama-2-7B-Chat-GGML llama2 model used](https://huggingface.co/TheBloke/Llama-2-7B-Chat-GGML)

app.py

@@ -20,7 +20,6 @@ def load_model():
 
 
 # request for blog content
-@st.cache_data
 def generate_blog(query, word_count, genre):
     """generate blog"""
 
@@ -28,11 +27,7 @@ def generate_blog(query, word_count, genre):
     As a you a blogger, please write a blog for the topic {query} and under gener {genre}.
     Please write the blog in {word_count} words.
     Your blog strictly should be in markdown language.
-    Also, come with an appropriate title for the blog and set it at the top in bold letters.
-    
-    Example: 
-    ## **YOUR TITLE**    
-    YOUR BLOG COME HERE     
+    Also, come with an appropriate title for the blog and set it at the top in bold letters.      
     """
 
     prompt = PromptTemplate(
@@ -46,42 +41,57 @@ def generate_blog(query, word_count, genre):
     return response
 
 
-st.set_page_config(
-    page_title="The Blogger",
-    page_icon="📝",
-    layout="centered",
-    initial_sidebar_state="collapsed",
-)
+def main():
+    "main"
 
-st.header("The Blogger 📝")
+    st.set_page_config(
+        page_title="The Blogger",
+        page_icon="📝",
+        layout="centered",
+        initial_sidebar_state="collapsed",
+    )
 
-user_input = st.text_input("Enter the topic you wanted the blog")
+    st.header("The Blogger 📝", divider="rainbow")
 
-col_1, col_2 = st.columns([5, 5])
-with col_1:
-    no_of_words = st.selectbox(
-        "Number of words", ("200", "250", "300", "350", "400", "500")
-    )
-with col_2:
-    blog_genres = st.selectbox(
-        "Genre of the blog",
-        (
-            "Business",
-            "Food",
-            "Music",
-            "Fitness",
-            "Travel",
-            "Gaming",
-            "Finance",
-            "Sports",
-            "Movie",
-            "Science & Technology",
-            "Lifestyle",
-        ),
-        index=0,
+    st.subheader(
+        "This application generates an :red[article] based on your :green[topic] and :blue[gener]. :sunglasses:"
     )
 
-submit = st.button("Blog")
-if submit:
-    blog = generate_blog(user_input, no_of_words, blog_genres)
-    st.write(blog)
+    st.markdown("[check out the repository](https://github.com/ThivaV/llama2_blogger)")
+
+    user_input = st.text_input("Enter the topic you wanted the blog")
+
+    col_1, col_2 = st.columns([5, 5])
+    with col_1:
+        no_of_words = st.selectbox(
+            "Number of words", ("200", "250", "300", "350", "400", "500")
+        )
+    with col_2:
+        blog_genres = st.selectbox(
+            "Genre of the blog",
+            (
+                "Science & Technology",
+                "Business",
+                "Food",
+                "Music",
+                "Fitness",
+                "Travel",
+                "Gaming",
+                "Finance",
+                "Sports",
+                "Movie",
+                "Lifestyle",
+            ),
+            index=0,
+        )
+
+    submit = st.button("Start Blogging")
+    if submit:
+        with st.spinner("Blogging started, Please wait.."):
+            blog = generate_blog(user_input, no_of_words, blog_genres)
+            st.write(blog)
+        st.success("Done!")
+
+
+if __name__ == "__main__":
+    main()