Update app.py

app.py

@@ -1,58 +1,67 @@
 import gradio as gr
 from pathlib import Path
-import os
 
-# Load models from file
 def load_models_from_file(filename):
     with open(filename, 'r') as f:
         return [line.strip() for line in f]
 
-# Initialize model
-models = load_models_from_file('models.txt')
-current_model = models[0]
+if __name__ == "__main__":
+    models = load_models_from_file('models.txt')
+    print(models)
 
-# Load Interfaces
+current_model = models[0]
 text_gen1 = gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link")
 models2 = [gr.Interface.load(f"models/{model}", live=True, preprocess=False) for model in models]
 
-# Function to generate text
 def text_it1(inputs):
     go_t1 = text_gen1(inputs)
     return go_t1
 
-# Function to set model
-def set_model(model_choice):
-    current_model = models[model_choice]
-    return f"{current_model}"
+def set_model(current_model):
+    current_model = models[current_model]
+    return gr.update(label=(f"{current_model}"))
 
-# Function to send input
 def send_it1(inputs, model_choice):
     proc1 = models2[model_choice]
     output1 = proc1(inputs)
-    
-    # Save image and prompt to file
-    prompt_safe_name = "".join(e for e in inputs if e.isalnum())
-    image_path = f"images/{prompt_safe_name}.png"
-    output1.save(image_path)
-    
-    with open('prompts.txt', 'a') as f:
-        f.write(f"{inputs}\n{image_path}\n")
-    
     return output1
 
-# Initialize Gradio Interface
-with gr.Blocks() as myface:
+css = ""
+
+with gr.Blocks(css=css) as myface:
+    with gr.Row():
+        with gr.Tab("Tools"):
+            with gr.Tab("View"):
+                with gr.Row():
+                    gr.Pil(label="Crop")
+                    gr.Pil(label="Crop")
+
+            with gr.Tab("Draw"):
+                with gr.Row():
+                    gr.Pil(label="Crop")
+                    gr.ImagePaint(label="Draw")
+                    gr.Textbox(label="", lines=8)
+                    gr.Textbox(label="", lines=8)
+
+            with gr.Tab("Color Picker"):
+                with gr.Row():
+                    gr.ColorPicker(label="Color")
+                    gr.ImagePaint(label="Draw")
+
     with gr.Row():
-        gr.Textbox(lines=4, label="Prompt").store("magic1")
-        gr.Dropdown(label="Select Model", choices=models, type="index", value=0, interactive=True).store("model_name1")
-        gr.Button("Generate Image").click(send_it1, inputs=["magic1", "model_name1"]).outputs("output1")
-        gr.Image(label="Image Output").store("output1")
-        gr.Textbox(label="Prompt Idea", lines=2).store("input_text")
-        gr.Button("Use Short Prompt").click(text_it1, inputs=["input_text"]).outputs("magic1")
-        
-    # Sidebar to show saved prompts
-    with open('prompts.txt', 'r') as f:
-        saved_prompts = f.readlines()
-    gr.Textbox(value=saved_prompts, lines=20, label="Saved Prompts", interactive=False, sidetab=True)
-    
-myface.launch()
+        gr.Textbox(lines=4, label="Magic Textbox")
+        gr.Button("Generate Image")
+
+    with gr.Row():
+        gr.Dropdown(label="Select Model", choices=[m for m in models], type="index", value=current_model)
+
+    with gr.Row():
+        gr.Image(label=(f"{current_model}"))
+
+    with gr.Row():
+        gr.Textbox(label="Prompt Idea", lines=2)
+        gr.Button("Use Short Prompt")
+        gr.Button("Extend Idea")
+
+    myface.queue(concurrency_count=200)
+    myface.launch(inline=True, show_api=False, max_threads=400)