Update app.py

app.py

@@ -24,11 +24,11 @@ NSFW_THRESHOLD = 0.5
 def generate_sketch(prompt, num_inference_steps, guidance_scale):
     # Classify the text for NSFW content
     text_classification = text_classifier(prompt)
-    print(text_classification)
+    
     # Check the classification results
     for result in text_classification:
         if result['label'] == 'nsfw' and result['score'] > NSFW_THRESHOLD: 
-            return "Inappropriate prompt detected. Please try another prompt."
+            return gr.update(visible=False),gr.Text(value="Inappropriate prompt detected. Please try another prompt.")
     
     image = pipe("sketched style, " + prompt, 
                  num_inference_steps=num_inference_steps, 
@@ -44,12 +44,12 @@ def generate_sketch(prompt, num_inference_steps, guidance_scale):
     # Check the classification results
     for result in image_classification:
         if result['label'] == 'nsfw' and result['score'] > NSFW_THRESHOLD: 
-            return "Inappropriate content detected. Please try another prompt."
+           return gr.update(visible=False),gr.Text(value="Inappropriate content detected. Please try another prompt.")
     
     image_path = "generated_sketch.png"
     
     image.save(image_path)
-    return image_path
+    return gr.Image(value=image_path), gr.update(visible=False)
 
 # Gradio interface with sliders for num_inference_steps and guidance_scale
 interface = gr.Interface(
@@ -59,10 +59,10 @@ interface = gr.Interface(
         gr.Slider(5, 50, value=24, step=1, label="Number of Inference Steps"),  # Slider for num_inference_steps
         gr.Slider(1.0, 10.0, value=3.5, step=0.1, label="Guidance Scale")  # Slider for guidance_scale
     ],
-    outputs="auto",
+    outputs=[gr.Image(), gr.Text()],
     title="Kids Sketch Generator",
     description="Enter a text prompt and generate a fun sketch for kids with customizable inference steps and guidance scale."
 )
 
 # Launch the app
-interface.launch()
+interface.launch()