Upload prompt.py

prompt.py

@@ -322,7 +322,7 @@ import os
 class HuggingFaceInferenceNode:
 
     def __init__(self):
-        self.client = InferenceClient("meta-llama/Meta-Llama-3.1-70B-Instruct")
+        self.client = InferenceClient("meta-llama/Meta-Llama-3.1-70B-Instruct") #
         self.prompts_dir = "./prompts"
         os.makedirs(self.prompts_dir, exist_ok=True)
 
@@ -379,7 +379,7 @@ You are allowed to make up film and branding names, and do them like 80's, 90's
                 {"role": "user", "content": user_message}
             ]
 
-            response = self.client.chat_completion(
+            response = self.client.chat_completion( #
                 model="meta-llama/Meta-Llama-3.1-70B-Instruct",
                 max_tokens=1024,
                 temperature=0.7,
@@ -412,160 +412,6 @@ pg_title = """<h1 align="center">FLUX Prompt Generator</h1>
 </center></p>
 """
 
-def create_interface():
-    prompt_generator = PromptGenerator()
-    huggingface_node = HuggingFaceInferenceNode()
-
-    with gr.Blocks(theme='bethecloud/storj_theme') as demo:
-        
-        gr.HTML(pg_title)
-
-        with gr.Row():
-            with gr.Column(scale=2):
-                with gr.Accordion("Basic Settings"):
-                    pg_seed = gr.Slider(0, 30000, label='Seed', step=1, value=random.randint(0,30000))
-                    pg_custom = gr.Textbox(label="Custom Input Prompt (optional)")
-                    pg_subject = gr.Textbox(label="Subject (optional)")
-                    
-                    # Add the radio button for global option selection
-                    pg_global_option = gr.Radio(
-                        ["Disabled", "Random", "No Figure Rand"],
-                        label="Set all options to:",
-                        value="Disabled"
-                    )
-                
-                with gr.Accordion("Artform and Photo Type", open=False):
-                    pg_artform = gr.Dropdown(["disabled", "random"] + ARTFORM, label="Artform", value="disabled")
-                    pg_photo_type = gr.Dropdown(["disabled", "random"] + PHOTO_TYPE, label="Photo Type", value="disabled")
-            
-                with gr.Accordion("Character Details", open=False):
-                    pg_body_types = gr.Dropdown(["disabled", "random"] + BODY_TYPES, label="Body Types", value="disabled")
-                    pg_default_tags = gr.Dropdown(["disabled", "random"] + DEFAULT_TAGS, label="Default Tags", value="disabled")
-                    pg_roles = gr.Dropdown(["disabled", "random"] + ROLES, label="Roles", value="disabled")
-                    pg_hairstyles = gr.Dropdown(["disabled", "random"] + HAIRSTYLES, label="Hairstyles", value="disabled")
-                    pg_clothing = gr.Dropdown(["disabled", "random"] + CLOTHING, label="Clothing", value="disabled")
-            
-                with gr.Accordion("Scene Details", open=False):
-                    pg_place = gr.Dropdown(["disabled", "random"] + PLACE, label="Place", value="disabled")
-                    pg_lighting = gr.Dropdown(["disabled", "random"] + LIGHTING, label="Lighting", value="disabled")
-                    pg_composition = gr.Dropdown(["disabled", "random"] + COMPOSITION, label="Composition", value="disabled")
-                    pg_pose = gr.Dropdown(["disabled", "random"] + POSE, label="Pose", value="disabled")
-                    pg_background = gr.Dropdown(["disabled", "random"] + BACKGROUND, label="Background", value="disabled")
-            
-                with gr.Accordion("Style and Artist", open=False):
-                    pg_additional_details = gr.Dropdown(["disabled", "random"] + ADDITIONAL_DETAILS, label="Additional Details", value="disabled")
-                    pg_photography_styles = gr.Dropdown(["disabled", "random"] + PHOTOGRAPHY_STYLES, label="Photography Styles", value="disabled")
-                    pg_device = gr.Dropdown(["disabled", "random"] + DEVICE, label="Device", value="disabled")
-                    pg_photographer = gr.Dropdown(["disabled", "random"] + PHOTOGRAPHER, label="Photographer", value="disabled")
-                    pg_artist = gr.Dropdown(["disabled", "random"] + ARTIST, label="Artist", value="disabled")
-                    pg_digital_artform = gr.Dropdown(["disabled", "random"] + DIGITAL_ARTFORM, label="Digital Artform", value="disabled")
-                
-                pg_generate_button = gr.Button("Generate Prompt")
-
-            with gr.Column(scale=2):
-                with gr.Accordion("Image and Caption", open=False):
-                    pg_input_image = gr.Image(label="Input Image (optional)")
-                    pg_caption_output = gr.Textbox(label="Generated Caption", lines=3)
-                    pg_create_caption_button = gr.Button("Create Caption")
-                    pg_add_caption_button = gr.Button("Add Caption to Prompt")
-
-                with gr.Accordion("Prompt Generation", open=True):
-                    pg_output = gr.Textbox(label="Generated Prompt / Input Text", lines=4)
-                    pg_t5xxl_output = gr.Textbox(label="T5XXL Output", visible=True)
-                    pg_clip_l_output = gr.Textbox(label="CLIP L Output", visible=True)
-                    pg_clip_g_output = gr.Textbox(label="CLIP G Output", visible=True)
-            
-            with gr.Column(scale=2):
-                with gr.Accordion("Prompt Generation with LLM", open=False):
-                    pg_model = gr.Dropdown(["Mixtral", "Mistral", "Llama 3", "Mistral-Nemo"], label="Model", value="Llama 3")
-                    pg_happy_talk = gr.Checkbox(label="Happy Talk", value=True)
-                    pg_compress = gr.Checkbox(label="Compress", value=True)
-                    pg_compression_level = gr.Radio(["soft", "medium", "hard"], label="Compression Level", value="hard")
-                    pg_poster = gr.Checkbox(label="Poster", value=False)
-                    pg_custom_base_prompt = gr.Textbox(label="Custom Base Prompt", lines=5)
-                pg_generate_text_button = gr.Button("Generate Prompt with LLM")
-                pg_text_output = gr.Textbox(label="Generated Text", lines=10)
-
-        def create_caption(image):
-            if image is not None:
-                return florence_caption(image)
-            return ""
-
-        pg_create_caption_button.click(
-            create_caption,
-            inputs=[pg_input_image],
-            outputs=[pg_caption_output]
-        )
-
-        pg_generate_button.click(
-            prompt_generator.generate_prompt,
-            inputs=[pg_seed, pg_custom, pg_subject, pg_artform, pg_photo_type, pg_body_types,
-                    pg_default_tags, pg_roles, pg_hairstyles,
-                    pg_additional_details, pg_photography_styles, pg_device, pg_photographer,
-                    pg_artist, pg_digital_artform,
-                    pg_place, pg_lighting, pg_clothing, pg_composition, pg_pose, pg_background],
-            outputs=[pg_output, gr.Number(visible=False), pg_t5xxl_output, pg_clip_l_output, pg_clip_g_output]
-        )
-
-        pg_add_caption_button.click(
-            prompt_generator.add_caption_to_prompt,
-            inputs=[pg_output, pg_caption_output],
-            outputs=[pg_output]
-        )
-
-        pg_generate_text_button.click(
-            huggingface_node.generate,
-            inputs=[pg_model, pg_output, pg_happy_talk, pg_compress, pg_compression_level,
-                    pg_poster, pg_custom_base_prompt],
-            outputs=pg_text_output
-        )
-
-        def update_all_options(choice):
-            updates = {}
-            if choice == "Disabled":
-                for dropdown in [
-                    pg_artform, pg_photo_type, pg_body_types, pg_default_tags,
-                    pg_roles, pg_hairstyles, pg_clothing,
-                    pg_place, pg_lighting, pg_composition, pg_pose, pg_background, pg_additional_details,
-                    pg_photography_styles, pg_device, pg_photographer, pg_artist, pg_digital_artform
-                ]:
-                    updates[dropdown] = gr.update(value="disabled")
-            elif choice == "Random":
-                for dropdown in [
-                    pg_artform, pg_photo_type, pg_body_types, pg_default_tags,
-                    pg_roles, pg_hairstyles, pg_clothing,
-                    pg_place, pg_lighting, pg_composition, pg_pose, pg_background, pg_additional_details,
-                    pg_photography_styles, pg_device, pg_photographer, pg_artist, pg_digital_artform
-                ]:
-                    updates[dropdown] = gr.update(value="random")
-            else:  # No Figure Random
-                for dropdown in [pg_photo_type, pg_body_types, pg_default_tags,
-                                 pg_roles, pg_hairstyles, pg_clothing, pg_pose, pg_additional_details]:
-                    updates[dropdown] = gr.update(value="disabled")
-                for dropdown in [pg_artform, pg_place, pg_lighting, pg_composition,
-                                 pg_background, pg_photography_styles, pg_device, pg_photographer,
-                                 pg_artist, pg_digital_artform]:
-                    updates[dropdown] = gr.update(value="random")
-            return updates
-        
-        pg_global_option.change(
-            update_all_options,
-            inputs=[pg_global_option],
-            outputs=[
-                pg_artform, pg_photo_type, pg_body_types, pg_default_tags,
-                pg_roles, pg_hairstyles, pg_clothing,
-                pg_place, pg_lighting, pg_composition, pg_pose, pg_background, pg_additional_details,
-                pg_photography_styles, pg_device, pg_photographer, pg_artist, pg_digital_artform
-            ]
-        )
-
-    return demo
-
-if __name__ == "__main__":
-    demo = create_interface()
-    demo.launch()
-
-
 def create_interface():
     prompt_generator = PromptGenerator()
     huggingface_node = HuggingFaceInferenceNode()
@@ -717,4 +563,4 @@ def create_interface():
 
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch()
+    demo.launch()