fixes

app.py

@@ -7,11 +7,11 @@ import matplotlib.pyplot as plt
 
 from PIL import Image
 from io import BytesIO
+from dotenv import load_dotenv
 from huggingface_hub import HfFileSystem, hf_hub_download
 
-
-def request_to_endpoint (request:dict, endpoint_id='fuxao2ukl3vx2z'):
-    api_key = "A6DTPK0VM02LEBYFSVJKKFZ9MTZQ4ED1CMBK0OE2"
+def request_to_endpoint (request:dict, endpoint_id='s4i4um7xakaq37'):
+    api_key = os.getenv("RUNPOD_API_KEY")
     url = f"https://api.runpod.ai/v2/{endpoint_id}/runsync"
     headers = {
         "accept": "application/json",
@@ -44,25 +44,25 @@ character_names, character_configs = prepare_characters()
 
 
 def process_input (name, scale, triggers, tech_prompt, tech_negative_prompt, prompts_count, *args):    
-    boxes = list(args)
-    prompts = []
-    negative_prompts = []
-
     # Если хотим задать свои триггерные слова или технический промпт
     triggers = triggers if triggers != "" else ",".join(character_configs[name]["trigger_words"])
     tech_prompt = tech_prompt if tech_prompt != "" else character_configs[name]["tech_prompt"]
     tech_negative_prompt = tech_negative_prompt if tech_negative_prompt != "" else character_configs[name]["tech_negative_prompt"]
+
     model = character_configs[name]["model"]
     model['loras'] = {name: scale}
     params = character_configs[name]["params"]
     params["cross_attention_kwargs"]['scale'] = scale
+
+    boxes = list(args)
+    prompts = []
+    negative_prompts = []
     for n in range(prompts_count):
-        prompts.append(f'{triggers}, {tech_prompt}, {boxes[2*n]}')
-        negative_prompts.append(f"{tech_negative_prompt}, {boxes[2*n + 1]}")
+        prompts.append(f'{triggers}, {tech_prompt}, {boxes[n]}')
+        negative_prompts.append(f"{tech_negative_prompt}")
     
     request_data = {
         "input": {
-            "mode": "inference",
             "model": model,
             "params": params,
             "prompt": prompts,
@@ -105,38 +105,42 @@ with gr.Blocks() as demo:
             visible=True,
         )
 
+        scale = gr.Slider(
+            minimum=0, 
+            maximum=2.0,
+            value=0.75, 
+            step=0.01, 
+            label="Selected LoRA scale:", 
+            interactive=True,
+        )
+
         with gr.Accordion(open=False):
-            scale = gr.Slider(0, 2.0, 0.75, step=0.01, label="Selected LoRA scale:", interactive=True)
             triggers = gr.Textbox(label=f"Trigger words:")
             tech_prompt = gr.Textbox(label=f"Technical prompt:")
             tech_negative_prompt = gr.Textbox(label=f"Negative technical prompt:")
 
     prompts_count = gr.State(1)
 
-    with gr.Row():
-        with gr.Column():
-            add_btn = gr.Button("Add prompt")
+    with gr.Group():
+        add_btn = gr.Button("Add prompt")
         del_btn = gr.Button("Delete prompt")
             
-    add_btn.click(lambda x: x + 1, prompts_count, prompts_count)
-    del_btn.click(lambda x: x - 1, prompts_count, prompts_count)
-
-    @gr.render(inputs=prompts_count)
-    def render_count(count):
-        boxes = []
-        for i in range(count):
-            with gr.Group():
-                prompt = gr.Textbox(key=i, label=f"Prompt {i+1}")
-                negative_prompt = gr.Textbox(key=i, label=f"Negative prompt {i+1}")
-
-            boxes.append(prompt)
-            boxes.append(negative_prompt)
-        
-        generate_btn.click(
-            process_input, 
-            [name, scale, triggers, tech_prompt, tech_negative_prompt, prompts_count]+boxes,
-            output
-        )
+        add_btn.click(lambda x: x + 1, prompts_count, prompts_count)
+        del_btn.click(lambda x: x - 1, prompts_count, prompts_count)
+
+        @gr.render(inputs=prompts_count)
+        def render_count(count):
+            boxes = []
+            for i in range(count):
+                with gr.Group():
+                    prompt = gr.Textbox(key=str(i), label=f"Prompt {i+1}")
+                boxes.append(prompt)
+            
+            generate_btn.click(
+                process_input, 
+                [name, scale, triggers, tech_prompt, tech_negative_prompt, prompts_count]+boxes,
+                output
+            )
 
     generate_btn = gr.Button("Generate!")