trial

app.py

@@ -23,12 +23,12 @@ def create_interface():
 
         with gr.Row():
             with gr.Column(scale=2):
-                with gr.Accordion("Basic Settings"):
+                with gr.Accordion("Settings"):
                     custom = gr.Textbox(label="Custom Input Prompt (optional)")
-                    
+
                 with gr.Accordion("Prompt Generation Options", open=False):
                     prompt_type = gr.Dropdown(
-                        choices=["Long", "Short", "Medium", "Long"],
+                        choices=["Long", "Short", "Medium", "OnlyObjects", "NoFigure", "Landscape", "Fantasy"],
                         label="Prompt Type",
                         value="Long",
                         interactive=True
@@ -45,14 +45,7 @@ def create_interface():
                     prompt_type.change(update_prompt_type, inputs=[prompt_type], outputs=[prompt_type])
 
             with gr.Column(scale=2):
-                generate_button = gr.Button("Generate Prompt")
-
-                with gr.Accordion("Generated Prompt", open=True):
-                    output = gr.Textbox(label="Generated Prompt", lines=4, show_copy_button=True)
-                    text_output = gr.Textbox(label="LLM Generated Text", lines=10, show_copy_button=True)
-            
-            with gr.Column(scale=2):
-                with gr.Accordion("""LLM Prompt Generation""", open=False):
+                with gr.Accordion("LLM Prompt Generation", open=False):
                     long_talk = gr.Checkbox(label="Long Talk", value=True)
                     compress = gr.Checkbox(label="Compress", value=True)
                     compression_level = gr.Dropdown(
@@ -72,8 +65,9 @@ def create_interface():
                 api_key = gr.Textbox(label="API Key", type="password", visible=False)
                 model = gr.Dropdown(label="Model", choices=[], value="")
 
-                generate_text_button = gr.Button("Generate Prompt with LLM")
-                text_output = gr.Textbox(label="Generated Text", lines=10, show_copy_button=True)
+                # **Single Button for Generating Prompt and Text**
+                generate_button = gr.Button("Generate Random Prompt with LLM")
+                text_output = gr.Textbox(label="LLM Generated Text", lines=10, show_copy_button=True)
 
         # Initialize Models based on provider
         def update_model_choices(provider):
@@ -91,42 +85,44 @@ def create_interface():
         llm_provider.change(update_model_choices, inputs=[llm_provider], outputs=[model])
         llm_provider.change(update_api_key_visibility, inputs=[llm_provider], outputs=[api_key])
 
-        # Generate Prompt Function
-        def generate_prompt(prompt_type, custom_input):
-            dynamic_seed = random.randint(0, 1000000)
-            result = llm_node.generate_prompt(dynamic_seed, prompt_type, custom_input)
-            return result
+        # **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):
+            try:
+                # Step 1: Generate Prompt
+                dynamic_seed = random.randint(0, 1000000)
+                prompt = llm_node.generate_prompt(dynamic_seed, prompt_type, custom_input)
+                print(f"Generated Prompt: {prompt}")
+
+                # Step 2: Generate Text with LLM
+                poster = False  # Set a default value or modify as needed
+                result = llm_node.generate(
+                    input_text=prompt,
+                    long_talk=long_talk,
+                    compress=compress,
+                    compression_level=compression_level,
+                    poster=poster,  # Added the missing 'poster' argument
+                    prompt_type=selected_prompt_type,
+                    custom_base_prompt=custom_base_prompt,
+                    provider=provider,
+                    api_key=api_key,
+                    model=model_selected
+                )
+                print(f"Generated Text: {result}")
 
-        generate_button.click(
-            generate_prompt,
-            inputs=[prompt_type, custom],
-            outputs=[output]
-        )
+                # Reset selected_prompt_type if necessary
+                selected_prompt_type = "Long" 
+                return result
 
-        # Generate Text with LLM
-        def generate_text_with_llm(output_prompt, long_talk, compress, compression_level, custom_base_prompt, provider, api_key, model_selected):
-            global selected_prompt_type
-            poster = False  # Set a default value or modify as needed
-            result = llm_node.generate(
-                input_text=output_prompt,
-                long_talk=long_talk,
-                compress=compress,
-                compression_level=compression_level,
-                poster=poster,  # Added the missing 'poster' argument
-                prompt_type=selected_prompt_type,
-                custom_base_prompt=custom_base_prompt,
-                provider=provider,
-                api_key=api_key,
-                model=model_selected
-            )
-            selected_prompt_type = "Long" 
-            return result
-
-        generate_text_button.click(
-            generate_text_with_llm,
-            inputs=[output, long_talk, compress, compression_level, custom_base_prompt, llm_provider, api_key, model],
+            except Exception as e:
+                print(f"An error occurred: {e}")
+                return f"Error occurred while processing the request: {str(e)}"
+
+        # **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],
             outputs=[text_output],
-            api_name="generate_text"
+            api_name="generate_random_prompt_with_llm"
         )
 
     return demo