Update app.py

app.py

@@ -154,8 +154,6 @@ class PipelineManager:
             return True, random.uniform(0.6, 0.8)
         
         return False, random.uniform(0.1, 0.3)
-    
-    def load_models(self) -> bool:
         """Load models with enhanced error handling and memory optimization"""
         if self.model_loaded:
             return True
@@ -715,4 +713,33 @@ def create_interface():
         
         img_generate_btn.click(
             fn=generate_img2img,
-            inputs=[img_input,
+            inputs=[img_input, img_prompt, img_negative, img_steps, img_guidance,
+                   img_strength, img_seed, img_quality],
+            outputs=[img_output_image, img_download_file, img_info],
+            show_progress=True
+        )
+        
+        # Example prompt buttons
+        txt_example_btn.click(fn=get_random_prompt, outputs=[txt_prompt])
+        img_example_btn.click(fn=get_random_prompt, outputs=[img_prompt])
+        
+        # Clear buttons
+        txt_clear_btn.click(lambda: "", outputs=[txt_prompt])
+        img_clear_btn.click(lambda: "", outputs=[img_prompt])
+    
+    return demo
+
+# Initialize and launch
+if __name__ == "__main__":
+    logger.info(f"🚀 Initializing CyberRealistic Pony Generator on {DEVICE}")
+    logger.info(f"📱 PyTorch version: {torch.__version__}")
+    logger.info(f"🛡️ NSFW Content Filter: Enabled")
+    
+    demo = create_interface()
+    demo.queue(max_size=20)  # Enable queuing for better UX
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True,
+        share=False  # Set to True if you want a public link
+    )