From e2930f9821c197da94e208b5ae73711002844efc Mon Sep 17 00:00:00 2001
From: Tony Beeman <beeman@gmail.com>
Date: Fri, 7 Oct 2022 17:46:39 -0700
Subject: Fix for Prompts_from_file showing extra textbox.

---
 modules/scripts.py | 30 ++++++++++++++++++++++++++----
 1 file changed, 26 insertions(+), 4 deletions(-)

(limited to 'modules')

diff --git a/modules/scripts.py b/modules/scripts.py
index 45230f9a..d8f87927 100644
--- a/modules/scripts.py
+++ b/modules/scripts.py
@@ -1,4 +1,5 @@
 import os
+from pydoc import visiblename
 import sys
 import traceback
 
@@ -31,6 +32,15 @@ class Script:
     def show(self, is_img2img):
         return True
 
+
+    # Called when the ui for this script has been shown.
+    # Useful for hiding some controls, since the scripts module sets visibility to
+    # everything to true. The parameters will be the parameters returned by the ui method
+    # The return value should be gradio updates, similar to what you would return
+    # from a Gradio event handler.
+    def on_show(self, *args):
+        return [ui.gr_show(True)] * len(args)
+
     # This is where the additional processing is implemented. The parameters include
     # self, the model object "p" (a StableDiffusionProcessing class, see
     # processing.py), and the parameters returned by the ui method.
@@ -125,20 +135,32 @@ class ScriptRunner:
             inputs += controls
             script.args_to = len(inputs)
 
-        def select_script(script_index):
+        def select_script(*args):
+            script_index = args[0]
+            on_show_updates = []
             if 0 < script_index <= len(self.scripts):
                 script = self.scripts[script_index-1]
                 args_from = script.args_from
                 args_to = script.args_to
+                script_args = args[args_from:args_to]
+                on_show_updates = wrap_call(script.on_show, script.filename, "on_show", *script_args)
             else:
                 args_from = 0
                 args_to = 0
 
-            return [ui.gr_show(True if i == 0 else args_from <= i < args_to) for i in range(len(inputs))]
+            ret = [ ui.gr_show(True)] # always show the dropdown
+            for i in range(1, len(inputs)):
+                if (args_from <= i < args_to):
+                    ret.append( on_show_updates[i - args_from] )
+                else:
+                    ret.append(ui.gr_show(False))
+            return ret
+
+            # return [ui.gr_show(True if (i == 0) else on_show_updates[i - args_from] if args_from <= i < args_to else False) for i in range(len(inputs))]
 
         dropdown.change(
             fn=select_script,
-            inputs=[dropdown],
+            inputs=inputs,
             outputs=inputs
         )
 
@@ -198,4 +220,4 @@ def reload_scripts(basedir):
     load_scripts(basedir)
 
     scripts_txt2img = ScriptRunner()
-    scripts_img2img = ScriptRunner()
+    scripts_img2img = ScriptRunner()
\ No newline at end of file
-- 
cgit v1.2.1