Update pages/Latest.py

pages/Latest.py

@@ -75,20 +75,24 @@ if 'titles' not in st.session_state:
 if 'selected_title' not in st.session_state:
     st.session_state.selected_title = None
 
-if st.button("Generate Titles"):
+if input_data:  # Only proceed if input_data is not empty
+    if st.button("Generate Titles"):
+        try:
+            topic = [HumanMessage(content=input_data)]
+            result = graph1.invoke({"topic": topic})
+            print(result)
+            st.session_state.titles = result.get('blog_topics', [])  # Avoid KeyError
+        except Exception as e:
+            st.error(str(e))
+
+if st.session_state.titles:
+    st.session_state.selected_title = st.selectbox("Pick the Title for your Blog", st.session_state.titles)
+
+if st.button("Generate Blog") and st.session_state.selected_title:
     try:
-        topic = [HumanMessage(content=input_data)]
-        result = graph1.invoke({"topic": topic})
-        print(result)
-        st.session_state.titles = result['blog_topics']
+        content = graph2.invoke({"final_topic": st.session_state.selected_title})
+        st.markdown(content.get('image_formated_blog', "No content generated."))  # Avoid KeyError
     except Exception as e:
         st.error(str(e))
 
-if st.session_state.titles is not None:
-    st.session_state.selected_title = st.selectbox("Pick the Title for your Blog", st.session_state.titles)
-
-if st.button("Generate Blog") and st.session_state.selected_title is not None:
-    content = graph2.invoke({"final_topic": st.session_state.selected_title})
-    st.markdown(content['image_formated_blog'])
-