Update app.py

app.py

@@ -38,12 +38,15 @@ def update_imgbox(choices):
     choices_plus = extend_choices(choices)
     return [gr.Image(None, label=m, visible=(m != 'NA')) for m in choices_plus]
 
-def gen_fn(model_str, prompt):
+def gen_fn(model_str, prompt, negative_prompt=None):
     if model_str == 'NA':
         return None
     noise = str('')  # Optional: str(randint(0, 99999999999))
     try:
-        result = models_load[model_str](f'{prompt} {noise}')
+        full_prompt = f'{prompt} {noise}'
+        if negative_prompt:
+            full_prompt += f' -{negative_prompt}'
+        result = models_load[model_str](full_prompt)
         # Check if result is an image or a file path
         if isinstance(result, str):  # Assuming result might be a file path
             if os.path.exists(result):
@@ -64,13 +67,14 @@ def gen_fn(model_str, prompt):
 def home():
     prompt = request.args.get('prompt', '')
     model = request.args.get('model', default_models[0] if default_models else 'NA')
+    negative_prompt = request.args.get('Nprompt', None)
     
     if not model or model not in models_load:
         return f'Invalid model: {model}', 400
     
     if prompt:
         # Generate the image
-        image = gen_fn(model, prompt)
+        image = gen_fn(model, prompt, negative_prompt)
         if isinstance(image, Image.Image):  # Ensure the result is a PIL image
             # Save image to BytesIO object
             img_io = BytesIO()
@@ -82,4 +86,4 @@ def home():
 
 if __name__ == '__main__':
     # Launch Flask app
-    app.run(host='0.0.0.0', port=7860)  # Run Flask app
+    app.run(host='0.0.0.0', port=7860)  # Run Flask app