Update app.py

app.py

@@ -52,6 +52,9 @@ st.write("Generate images based on rough sketches, text input, or both.")
 
 option = st.radio("Select Input Type", ("Sketch", "Text", "Both"))
 
+sketch_file = None
+text_input = None
+
 if option in ["Sketch", "Both"]:
     sketch_file = st.file_uploader("Upload a Sketch", type=["png", "jpg", "jpeg"])
 
@@ -59,14 +62,18 @@ if option in ["Text", "Both"]:
     text_input = st.text_input("Enter Text Prompt", placeholder="Describe the image you want to generate")
 
 if st.button("Generate"):
+    # Ensure text_input and sketches are handled properly
+    sketches = BytesIO(sketch_file.read()) if sketch_file else None
+    
     if option == "Sketch" and not sketch_file:
         st.error("Please upload a sketch.")
     elif option == "Text" and not text_input:
         st.error("Please provide text input.")
+    elif option == "Both" and not (sketch_file or text_input):
+        st.error("Please provide both a sketch and a text prompt.")
     else:
         # Generate images based on user input
         with st.spinner("Generating images..."):
-            sketches = BytesIO(sketch_file.read()) if sketch_file else None
             images = generate_images(pipe, text_input=text_input, sketch=sketches)
 
         if images: