Update app.py

app.py

@@ -1,4 +1,4 @@
-'''
+
 import gradio as gr
 import os
 from random import randint
@@ -246,171 +246,4 @@ demo.launch(max_threads=200)
 
 #load_fn(models)
 #num_models = len(models)
-#default_models = models[:num_models]
-'''
-
-import gradio as gr
-import os
-from random import randint
-from all_models import models
-from datetime import datetime
-from concurrent.futures import TimeoutError, ThreadPoolExecutor
-import logging
-import traceback
-
-# Disable GPU usage
-os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
-
-# Logging setup
-logging.basicConfig(level=logging.WARNING)
-now2 = 0
-index_tracker = 0 
-model_scores = {model: 0 for model in models}   
-processed_models_count = 0
-kii = "blonde mohawk femboy playing game with self at computer with programmer socks on, still a wip"
-combined_prompt = ""
-
-# Function to get current timestamp
-def get_current_time():
-    now = datetime.now()
-    current_time = now.strftime("%Y-%m-%d %H:%M:%S")
-    return f'{kii} {current_time}'
-
-# Function to sanitize file names
-def sanitize_file_name(file_name, max_length=100):
-    return file_name[:max_length].replace(" ", "_").replace("/", "_")
-
-# Load models
-def load_fn(models):
-    global models_load
-    models_load = {}
-    for model in models:
-        try:
-            m = gr.load(f'models/{model}')
-            if m is not None:
-                models_load[model] = m
-                print(f"Loaded model: {model}")
-            else:
-                print(f"Warning: Model {model} could not be loaded.")
-                models_load[model] = None
-        except Exception as error:
-            print(f"Error loading model {model}: {error}")
-            models_load[model] = None
-
-load_fn(models)
-num_models = len(models)
-default_models = models[:num_models]
-
-# Extend choices to match model count
-def extend_choices(choices):
-    return choices + (num_models - len(choices)) * ['NA']
-
-def update_imgbox(choices):
-    choices_plus = extend_choices(choices)
-    return [gr.Image(None, label=m, visible=(m != 'NA')) for m in choices_plus]
-
-executor = ThreadPoolExecutor(max_workers=num_models)
-
-def gen_fn(model_str, prompt):
-    global index_tracker, model_scores, processed_models_count 
-    if model_str == 'NA' or models_load.get(model_str) is None:
-        return None  # Avoid calling an undefined function
-    
-    try:
-        index_tracker = (index_tracker + 1) % len(models)
-        current_model_index = index_tracker
-        current_model_name = models[current_model_index]
-        
-        max_prompt_length = 100
-        truncated_prompt = sanitize_file_name(prompt[:max_prompt_length])
-        combined_prompt = f"{truncated_prompt}_{randint(0, 9999)}"
-
-        # Execute model with timeout
-        future = executor.submit(models_load[model_str], f"{combined_prompt}")
-        response = future.result(timeout=150)
-
-        if isinstance(response, gr.Image):
-            return response
-        elif isinstance(response, str): 
-            model_scores[current_model_name] += 1
-            processed_models_count += 1
-            if processed_models_count == len(models):
-                processed_models_count = 0
-            return response
-
-    except TimeoutError:
-        processed_models_count += 1
-        if processed_models_count == len(models):
-            processed_models_count = 0
-        return None
-
-    except Exception as e:
-        processed_models_count += 1
-        if processed_models_count == len(models):
-            processed_models_count = 0
-        return None
-
-def make_me():
-    with gr.Row():
-        txt_input = gr.Textbox(lines=2, value=kii, label=None)
-        gen_button = gr.Button('Generate images')
-        stop_button = gr.Button('Stop', variant='secondary', interactive=False)
-    #    gen_button.click(lambda _: gr.update(interactive=True), None, stop_button)
-        gen_button.click(lambda s: gr.update(interactive=True), inputs=[], outputs=[stop_button])
-
-    with gr.Row():
-        output = [gr.Image(label=m) for m in default_models]
-        current_models = [gr.Textbox(m, visible=False) for m in default_models]
-        for m, o in zip(current_models, output):
-            print(f"Connecting {m} to gen_fn")  # Debugging output
-            gen_event = gen_button.click(gen_fn, [m, txt_input], o)
-            stop_button.click(lambda s: gr.update(interactive=False), inputs=[], outputs=[stop_button], cancels=[gen_event])
-
-
-    with gr.Accordion('Model selection', visible=False):
-        model_choice = gr.CheckboxGroup(models, label=f'{num_models} models selected', value=default_models, interactive=True)
-        model_choice.change(update_imgbox, model_choice, output)
-        model_choice.change(extend_choices, model_choice, current_models)
-
-# JavaScript to fix scrolling issue
-js_code = """
-<script>
-const originalScroll = window.scrollTo;
-gradio.Toast.show = function() {
-    originalShowToast.apply(this, arguments);
-    window.scrollTo = function() {};
-};
-setTimeout(() => {
-    window.scrollTo = originalScroll;
-}, 1000);
-</script>
-"""
-
-# Launch Gradio App
-with gr.Blocks(css="""
-textarea:hover { background:#55555555; }
-textarea { font-size: 1.5em; letter-spacing: 3px; color: limegreen; }
-""") as demo:
-    gr.Markdown("<script>" + js_code + "</script>")
-    make_me()
-
-demo.queue()
-demo.config["queue"] = False
-
-if __name__ == "__main__":
-    print("Launching Gradio app...")
-    demo.launch(debug=True, share=False, max_threads=200)
-
-
-
-            
-
-
-
-
-
-
-
-
-    
-        
+#default_models = models[:num_models]