global

app.py

@@ -10,9 +10,6 @@ title = """<h1 align="center">Random Prompt Generator</h1>
 </center></p>
 """
 
-# Global variable to store selected prompt type
-selected_prompt_type = "Long"  # Default value
-
 def create_interface():
     llm_node = LLMInferenceNode()
 
@@ -31,15 +28,16 @@ def create_interface():
                     interactive=True
                 )
                 
-                # Function to update the selected prompt type
-                def update_prompt_type(value):
-                    global selected_prompt_type
-                    selected_prompt_type = value
-                    print(f"Updated prompt type: {selected_prompt_type}")
-                    return value
+                # Add a State component to store the selected prompt type
+                prompt_type_state = gr.State("Long")
+
+                # Update the function to use State
+                def update_prompt_type(value, state):
+                    print(f"Updated prompt type: {value}")
+                    return value, value
                 
                 # Connect the update_prompt_type function to the prompt_type dropdown
-                prompt_type.change(update_prompt_type, inputs=[prompt_type], outputs=[prompt_type])
+                prompt_type.change(update_prompt_type, inputs=[prompt_type, prompt_type_state], outputs=[prompt_type, prompt_type_state])
 
             with gr.Column(scale=2):
                 with gr.Accordion("LLM Prompt Generation", open=False):
@@ -105,8 +103,7 @@ def create_interface():
         )
 
         # **Unified Function to Generate Prompt and Text**
-        def generate_random_prompt_with_llm(custom_input, prompt_type, long_talk, compress, compression_level, custom_base_prompt, provider, api_key, model_selected):
-            global selected_prompt_type  # Declare as global
+        def generate_random_prompt_with_llm(custom_input, prompt_type, long_talk, compress, compression_level, custom_base_prompt, provider, api_key, model_selected, prompt_type_state):
             try:
                 # Step 1: Generate Prompt
                 dynamic_seed = random.randint(0, 1000000)
@@ -128,7 +125,7 @@ def create_interface():
                     compress=compress,
                     compression_level=compression_level,
                     poster=poster,  # Added the missing 'poster' argument
-                    prompt_type=selected_prompt_type,
+                    prompt_type=prompt_type_state,
                     custom_base_prompt=custom_base_prompt,
                     provider=provider,
                     api_key=api_key,
@@ -136,8 +133,6 @@ def create_interface():
                 )
                 print(f"Generated Text: {result}")
 
-                # Reset selected_prompt_type if necessary
-                selected_prompt_type = "Long" 
                 return result
 
             except Exception as e:
@@ -147,7 +142,7 @@ def create_interface():
         # **Connect the Unified Function to the Single Button**
         generate_button.click(
             generate_random_prompt_with_llm,
-            inputs=[custom, prompt_type, long_talk, compress, compression_level, custom_base_prompt, llm_provider, api_key, model],
+            inputs=[custom, prompt_type, long_talk, compress, compression_level, custom_base_prompt, llm_provider, api_key, model, prompt_type_state],
             outputs=[text_output],
             api_name="generate_random_prompt_with_llm"
         )