Update app.py

app.py

@@ -75,67 +75,42 @@ def gen_fn(model_str, prompt):
         response = models_load[model_str](f'{combined_prompt}')
         return model_name, response
 
-    try:
-        # Run the image generation task with a timeout
-        future = executor.submit(generate_image)
-        #current_model_name, 
-        response  = future.result(timeout=100)  # Set timeout in seconds
-
-        if isinstance(response, str):  # If the response is valid
-         #   print(f"84{current_model_name}")
-            print(f"85 this should pin the image ")
-            print(f"86{response}")
-            if processed_models_count == 0:
-                print(f"**************")
-                print(f"{prompt}")
-                print(f"{prompt}")
-                print(f"{prompt}")
-                print(f"**************")
-            #model_scores[current_model_name] += 1
-            #print(f"OOO n:{processed_models_count} x:{current_model_index}  r[{model_scores[current_model_name]}] {model_str}")
-            #processed_models_count += 1
-            if processed_models_count == len(models):
-               print(f" **** ME facepalm")
-          #      print("\nCycle Complete! Updated Scores:")
-          #      print(model_scores)
-               processed_models_count = 0
-            print(f"100 RIIIIIIIght cheer ")    
+        try:
+        index_tracker = (index_tracker + 1) % len(models)
+        current_model_index = index_tracker
+        current_model_name = models[current_model_index]
+        combined_prompt = f"{prompt} {randint(0, 9999)}"
+        response = models_load[model_str](f"{combined_prompt}")
+ 
+        if isinstance(response, gr.Image):
             return response
-
-        else:
-            print(f"elsed out")
-            print(f"hang da bish")
-            processed_models_count += 1
+        elif isinstance(response, tuple):
             return None
-
-    except TimeoutError:
-        if processed_models_count == 0:
+        elif isinstance(response, str):  # If the response is a path or URL, pass it as a string
+            if processed_models_count == 0:
                 print(f"**************")
                 print(f"{prompt}")
                 print(f"{prompt}")
                 print(f"{prompt}")
                 print(f"**************")
-        # model_score = model_scores.get(current_model_name, 0) if current_model_name else 0
-        model_score = model_scores.get(current_model_name, 0)
-        print(f"TTT n:{processed_models_count} x:{index_tracker}  r[{model_score}] {model_str}")
-        processed_models_count += 1
-        if processed_models_count == len(models):
+            model_scores[current_model_name] += 1
+            print(f"OOO n:{processed_models_count} x:{current_model_index}  r[{model_scores[current_model_name]}] {model_str}")
+            processed_models_count += 1
+            if processed_models_count == len(models):
                 print("\nCycle Complete! Updated Scores:")
                 print(model_scores)
                 processed_models_count = 0
-        return None
+            return response
 
+            
     except Exception as e:
-        print(f"{e}")
         if processed_models_count == 0:
                 print(f"**************")
                 print(f"{prompt}")
                 print(f"{prompt}")
                 print(f"{prompt}")
                 print(f"**************")
-       # model_score = model_scores.get(current_model_name, 0) if current_model_name else 0
-        model_score = model_scores.get(current_model_name, 0)
-        print(f"--- n:{processed_models_count} x:{index_tracker}  r[{model_score}] {model_str}")
+        print(f"--- n:{processed_models_count} x:{current_model_index}  r[{model_scores[current_model_name]}] {model_str}")
         processed_models_count += 1
         if processed_models_count == len(models):
                 print("\nCycle Complete! Updated Scores:")
@@ -146,10 +121,6 @@ def gen_fn(model_str, prompt):
 
 
 
-
-
-
-
     
 def make_me():
     with gr.Row():