Update pages/Custom.py

pages/Custom.py

@@ -47,10 +47,13 @@ graph = builder.compile()
 st.title("Generate Custom Title Blog")
 input_data = st.text_input("Enter a Title for your blog:")
 
-
-if st.button("Generate Blog"):
-    content = graph.invoke({"final_topic": input_data})
-    st.markdown(content['image_formated_blog'])
+if input_data:  # Ensure input_data is not empty
+    if st.button("Generate Blog"):
+        try:
+            content = graph.invoke({"final_topic": input_data})
+            st.markdown(content.get('image_formated_blog', "No content generated."))  # Avoid KeyError
+        except Exception as e:
+            st.error(str(e))