Update app.py

app.py

@@ -18,54 +18,68 @@ def save_ideas(ideas):
     with open("ideas.json", "w") as file:
         json.dump(ideas, file)
 
+# Function to generate content
+def generate_content(topic):
+    hub_llm = HuggingFaceHub(repo_id="HuggingFaceH4/zephyr-7b-beta")
+    prompt = PromptTemplate(
+        input_variables=['keyword'],
+        template="""
+        Write a comprehensive article about {keyword} covering the following aspects:
+        Introduction, History and Background, Key Concepts and Terminology, Use Cases and Applications, Benefits and Drawbacks, Future Outlook, Conclusion
+        Ensure that the article is well-structured, informative, and at least 1500 words long. Use SEO best practices for content optimization.
+        """
+    )
+    hub_chain = LLMChain(prompt=prompt, llm=hub_llm, verbose=True)
+    content = hub_chain.run(topic)
 
-topic = st.text_input("Enter Topic for the bog")
-button_clicked = st.button("Create blog!")
+    subheadings = [
+        "Introduction",
+        "History and Background",
+        "Key Concepts and Terminology",
+        "Use Cases and Applications",
+        "Benefits and Drawbacks",
+        "Future Outlook",
+        "Conclusion",
+    ]
 
-existing_ideas = load_ideas()
-st.sidebar.header("Previous Ideas:")
+    for subheading in subheadings:
+        if (subheading + ":") in content:
+            content = content.replace(subheading + ":", "## " + subheading + "\n")
+        elif subheading in content:
+            content = content.replace(subheading, "## " + subheading + "\n")
 
-markdown_placeholder = st.empty()
+    return content
 
-while(True):
-  keys = list(set([key for idea in existing_ideas for key in idea.keys()]))
-  if keys:  
-    selected_idea = st.sidebar.selectbox("Select Idea", keys, key=f"selectbox_{len(keys)}")
-    selected_idea_from_list = next((idea for idea in existing_ideas if selected_idea in idea), None)
-    markdown_placeholder.markdown(selected_idea_from_list[selected_idea])
-    if button_clicked:
+# Streamlit app
+st.title("Blog Generator")
 
-        hub_llm = HuggingFaceHub(repo_id ="HuggingFaceH4/zephyr-7b-beta")
-        prompt = PromptTemplate(
-            input_variables = ['keyword'],
-            template = """
-            Write a comprehensive article about {keyword} covering the following aspects:
-            Introduction, History and Background, Key Concepts and Terminology, Use Cases and Applications, Benefits and Drawbacks, Future Outlook, Conclusion
-            Ensure that the article is well-structured, informative, and at least 1500 words long. Use SEO best practices for content optimization.
-            """)
-        hub_chain = LLMChain(prompt=prompt,llm = hub_llm,verbose=True)
+# Input and button
+topic = st.text_input("Enter Title for the blog")
+button_clicked = st.button("Create blog!")
 
-        content = hub_chain.run(topic)
+# Load existing ideas
+existing_ideas = load_ideas()
+st.sidebar.header("Previous Ideas:")
 
+# Display existing ideas in the sidebar
+keys = list(set([key for idea in existing_ideas for key in idea.keys()]))
+selected_idea = st.sidebar.selectbox("Select Idea", keys, key="selectbox", index=0)
 
-        subheadings = [
-                "Introduction:",
-                "History and Background:",
-                "Key Concepts and Terminology:",
-                "Use Cases and Applications:",
-                "Benefits and Drawbacks:",
-                "Future Outlook:",
-                "Conclusion:",
-            ]
-        organized_content = ""
+# Display content for the selected idea
+selected_idea_from_list = next((idea for idea in existing_ideas if selected_idea in idea), None)
+st.markdown(selected_idea_from_list[selected_idea])
 
-        for subheading in subheadings:
-            if subheading in content:
-                content = content.replace(subheading,"## "+subheading+"\n")
+# Handle button click
+if button_clicked:
+    # Generate content and update existing ideas
+    content = generate_content(topic)
+    existing_ideas.append({topic: content})
+    save_ideas(existing_ideas)
 
-        existing_ideas.append({topic:content})
-        save_ideas(existing_ideas)
-        keys = list(set([key for idea in existing_ideas for key in idea.keys()]))
-        st.sidebar.selectbox("Select Idea", keys, key=f"selectbox_{len(keys)}")
-        markdown_placeholder.markdown(content)
+    # Update keys and selected idea in the sidebar
+    keys = list(set([key for idea in existing_ideas for key in idea.keys()]))
+    selected_idea = st.sidebar.selectbox("Select Idea", keys, key="selectbox", index=keys.index(topic))
 
+    # Update selected idea content
+    selected_idea_from_list = next((idea for idea in existing_ideas if selected_idea in idea), None)
+    st.markdown(selected_idea_from_list[selected_idea])