Revert "Merge pull request #7595 from siutin/feature/restore-progress"

This reverts commit 80987c36f9, reversing
changes made to 2e78e65a22.

javascript/hints.js

@@ -22,7 +22,7 @@ titles = {
     "\u{1f4cb}": "Apply selected styles to current prompt",
     "\u{1f4d2}": "Paste available values into the field",
     "\u{1f3b4}": "Show/hide extra networks",
-		"\u{1F300}": "Restore progress",
+
     "Inpaint a part of image": "Draw a mask over an image, and the script will regenerate the masked area with content according to prompt",
     "SD upscale": "Upscale image normally, split result into tiles, improve each tile using img2img, merge whole image back",
 

javascript/ui.js

@@ -362,32 +362,6 @@ function selectCheckpoint(name){
     gradioApp().getElementById('change_checkpoint').click()
 }
 
-function restoreProgress (task_tag) {
-
-  if (task_tag) {
-    let successHandler = ({ current_task }) => {
-      if (current_task) {
-        showSubmitButtons(task_tag, false)
-        requestProgress(current_task, gradioApp().getElementById(`${task_tag}_gallery_container`), gradioApp().getElementById(`${task_tag}_gallery`), function(){
-          showSubmitButtons(task_tag, true)
-        })
-      }
-    }
-
-    let errorHandler = e => window.alert(`invalid internal api respsonse. message: ${e}`)
-
-    fetch("./internal/current_task")
-      .then(res => res.json())
-      .then(successHandler)
-      .catch(errorHandler)
-  }
-
-  var res = create_submit_args(arguments)
-  res[0] = 0
-  return res
-
-}
-
 function currentImg2imgSourceResolution(_, _, scaleBy){
     var img = gradioApp().querySelector('#mode_img2img > div[style="display: block;"] img')
     return img ? [img.naturalWidth, img.naturalHeight, scaleBy] : [0, 0, scaleBy]
@@ -403,4 +377,3 @@ function updateImg2imgResizeToTextAfterChangingImage(){
 
     return []
 }
-

modules/call_queue.py

@@ -4,16 +4,10 @@ import threading
 import traceback
 import time
 
-import gradio as gr
 from modules import shared, progress
 
 queue_lock = threading.Lock()
-queue_lock_condition = threading.Condition(lock=queue_lock)
 
-def wrap_session_call(func):
-  def f(request: gr.Request, *args, **kwargs):
-    return func(request, *args, **kwargs)
-  return f
 
 def wrap_queued_call(func):
     def f(*args, **kwargs):
@@ -26,31 +20,29 @@ def wrap_queued_call(func):
 
 
 def wrap_gradio_gpu_call(func, extra_outputs=None):
-    def f(request: gr.Request, *args, **kwargs):
+    def f(*args, **kwargs):
-        user = request.username
 
         # if the first argument is a string that says "task(...)", it is treated as a job id
         if len(args) > 0 and type(args[0]) == str and args[0][0:5] == "task(" and args[0][-1] == ")":
             id_task = args[0]
-            progress.add_task_to_queue(user, id_task)
+            progress.add_task_to_queue(id_task)
         else:
             id_task = None
 
         with queue_lock:
             shared.state.begin()
-            progress.start_task(user, id_task)
+            progress.start_task(id_task)
 
             try:
                 res = func(*args, **kwargs)
             finally:
-                progress.finish_task(user, id_task)
+                progress.finish_task(id_task)
-                progress.set_last_task_result(user, id_task, res)
 
             shared.state.end()
 
         return res
 
-    return wrap_session_call(wrap_gradio_call(f, extra_outputs=extra_outputs, add_stats=True))
+    return wrap_gradio_call(f, extra_outputs=extra_outputs, add_stats=True)
 
 
 def wrap_gradio_call(func, extra_outputs=None, add_stats=False):

modules/progress.py

@@ -4,84 +4,38 @@ import time
 
 import gradio as gr
 from pydantic import BaseModel, Field
-from typing import Optional
-from fastapi import Depends, Security
-from fastapi.security import APIKeyCookie
 
-from modules import call_queue
 from modules.shared import opts
 
 import modules.shared as shared
 
 
-current_task_user = None
 current_task = None
 pending_tasks = {}
 finished_tasks = []
 
 
-def start_task(user, id_task):
+def start_task(id_task):
     global current_task
-    global current_task_user
 
-    current_task_user = user
     current_task = id_task
-    pending_tasks.pop((user, id_task), None)
+    pending_tasks.pop(id_task, None)
 
 
-def finish_task(user, id_task):
+def finish_task(id_task):
     global current_task
-    global current_task_user
 
     if current_task == id_task:
         current_task = None
 
-    if current_task_user == user:
+    finished_tasks.append(id_task)
-        current_task_user = None
-
-    finished_tasks.append((user, id_task))
     if len(finished_tasks) > 16:
         finished_tasks.pop(0)
 
 
-def add_task_to_queue(user, id_job):
+def add_task_to_queue(id_job):
-    pending_tasks[(user, id_job)] = time.time()
+    pending_tasks[id_job] = time.time()
 
-last_task_id = None
-last_task_result = None
-last_task_user = None
-
-def set_last_task_result(user, id_job, result):
-
-  global last_task_id
-  global last_task_result
-  global last_task_user
-
-  last_task_id = id_job
-  last_task_result = result
-  last_task_user = user
-
-
-def restore_progress_call(request: gr.Request):
-    if current_task is None:
-
-      # image, generation_info, html_info, html_log
-      return tuple(list([None, None, None, None]))
-
-    else:
-      user = request.username
-
-      if current_task_user == user:
-        t_task = current_task
-        with call_queue.queue_lock_condition:
-          call_queue.queue_lock_condition.wait_for(lambda: t_task == last_task_id)
-
-        return last_task_result
-
-      return tuple(list([None, None, None, None]))
-
-class CurrentTaskResponse(BaseModel):
-  current_task: str = Field(default=None, title="Task ID", description="id of the current progress task")
 
 class ProgressRequest(BaseModel):
     id_task: str = Field(default=None, title="Task ID", description="id of the task to get progress for")
@@ -102,21 +56,6 @@ class ProgressResponse(BaseModel):
 def setup_progress_api(app):
     return app.add_api_route("/internal/progress", progressapi, methods=["POST"], response_model=ProgressResponse)
 
-def setup_current_task_api(app):
-
-    def get_current_user(token: Optional[str] = Security(APIKeyCookie(name="access-token", auto_error=False))):
-      return None if token is None else app.tokens.get(token)
-
-    def current_task_api(current_user: str = Depends(get_current_user)):
-
-      if app.auth is None or current_task_user == current_user:
-        current_user_task = current_task
-      else:
-        current_user_task = None
-
-      return CurrentTaskResponse(current_task=current_user_task)
-
-    return app.add_api_route("/internal/current_task", current_task_api, methods=["GET"], response_model=CurrentTaskResponse)
 
 def progressapi(req: ProgressRequest):
     active = req.id_task == current_task
@@ -156,4 +95,5 @@ def progressapi(req: ProgressRequest):
     else:
         live_preview = None
 
-    return ProgressResponse(active=active, queued=queued, completed=completed, progress=progress, eta=eta, live_preview=live_preview, id_live_preview=id_live_preview, textinfo=shared.state.textinfo)
+    return ProgressResponse(active=active, queued=queued, completed=completed, progress=progress, eta=eta, live_preview=live_preview, id_live_preview=id_live_preview, textinfo=shared.state.textinfo)
+

modules/ui.py

@@ -41,7 +41,6 @@ from modules.textual_inversion import textual_inversion
 import modules.hypernetworks.ui
 from modules.generation_parameters_copypaste import image_from_url_text
 import modules.extras
-from modules.progress import restore_progress_call
 
 warnings.filterwarnings("default" if opts.show_warnings else "ignore", category=UserWarning)
 
@@ -82,7 +81,6 @@ apply_style_symbol = '\U0001f4cb'  # 📋
 clear_prompt_symbol = '\U0001f5d1\ufe0f'  # 🗑️
 extra_networks_symbol = '\U0001F3B4'  # 🎴
 switch_values_symbol = '\U000021C5' # ⇅
-restore_progress_symbol = '\U0001F300'  # 🌀
 
 
 def plaintext_to_html(text):
@@ -327,7 +325,6 @@ def create_toprow(is_img2img):
                 extra_networks_button = ToolButton(value=extra_networks_symbol, elem_id=f"{id_part}_extra_networks")
                 prompt_style_apply = ToolButton(value=apply_style_symbol, elem_id=f"{id_part}_style_apply")
                 save_style = ToolButton(value=save_style_symbol, elem_id=f"{id_part}_style_create")
-                restore_progress_button = ToolButton(value=restore_progress_symbol, elem_id=f"{id_part}_restore_progress")
 
                 token_counter = gr.HTML(value="<span>0/75</span>", elem_id=f"{id_part}_token_counter", elem_classes=["token-counter"])
                 token_button = gr.Button(visible=False, elem_id=f"{id_part}_token_button")
@@ -345,7 +342,7 @@ def create_toprow(is_img2img):
                 prompt_styles = gr.Dropdown(label="Styles", elem_id=f"{id_part}_styles", choices=[k for k, v in shared.prompt_styles.styles.items()], value=[], multiselect=True)
                 create_refresh_button(prompt_styles, shared.prompt_styles.reload, lambda: {"choices": [k for k, v in shared.prompt_styles.styles.items()]}, f"refresh_{id_part}_styles")
 
-    return prompt, prompt_styles, negative_prompt, submit, button_interrogate, button_deepbooru, prompt_style_apply, save_style, paste, extra_networks_button, token_counter, token_button, negative_token_counter, negative_token_button, restore_progress_button
+    return prompt, prompt_styles, negative_prompt, submit, button_interrogate, button_deepbooru, prompt_style_apply, save_style, paste, extra_networks_button, token_counter, token_button, negative_token_counter, negative_token_button
 
 
 def setup_progressbar(*args, **kwargs):
@@ -462,7 +459,7 @@ def create_ui():
     modules.scripts.scripts_txt2img.initialize_scripts(is_img2img=False)
 
     with gr.Blocks(analytics_enabled=False) as txt2img_interface:
-        txt2img_prompt, txt2img_prompt_styles, txt2img_negative_prompt, submit, _, _, txt2img_prompt_style_apply, txt2img_save_style, txt2img_paste, extra_networks_button, token_counter, token_button, negative_token_counter, negative_token_button, restore_progress_button = create_toprow(is_img2img=False)
+        txt2img_prompt, txt2img_prompt_styles, txt2img_negative_prompt, submit, _, _, txt2img_prompt_style_apply, txt2img_save_style, txt2img_paste, extra_networks_button, token_counter, token_button, negative_token_counter, negative_token_button = create_toprow(is_img2img=False)
 
         dummy_component = gr.Label(visible=False)
         txt_prompt_img = gr.File(label="", elem_id="txt2img_prompt_image", file_count="single", type="binary", visible=False)
@@ -594,18 +591,6 @@ def create_ui():
 
             res_switch_btn.click(lambda w, h: (h, w), inputs=[width, height], outputs=[width, height], show_progress=False)
 
-            restore_progress_button.click(
-              fn=restore_progress_call,
-              _js="() => restoreProgress('txt2img')",
-              inputs=[],
-              outputs=[
-                  txt2img_gallery,
-                  generation_info,
-                  html_info,
-                  html_log,
-              ]
-            )
-
             txt_prompt_img.change(
                 fn=modules.images.image_data,
                 inputs=[
@@ -674,7 +659,7 @@ def create_ui():
     modules.scripts.scripts_img2img.initialize_scripts(is_img2img=True)
 
     with gr.Blocks(analytics_enabled=False) as img2img_interface:
-        img2img_prompt, img2img_prompt_styles, img2img_negative_prompt, submit, img2img_interrogate, img2img_deepbooru, img2img_prompt_style_apply, img2img_save_style, img2img_paste, extra_networks_button, token_counter, token_button, negative_token_counter, negative_token_button, restore_progress_button = create_toprow(is_img2img=True)
+        img2img_prompt, img2img_prompt_styles, img2img_negative_prompt, submit, img2img_interrogate, img2img_deepbooru, img2img_prompt_style_apply, img2img_save_style, img2img_paste, extra_networks_button, token_counter, token_button, negative_token_counter, negative_token_button = create_toprow(is_img2img=True)
 
         img2img_prompt_img = gr.File(label="", elem_id="img2img_prompt_image", file_count="single", type="binary", visible=False)
 
@@ -966,18 +951,6 @@ def create_ui():
             submit.click(**img2img_args)
             res_switch_btn.click(lambda w, h: (h, w), inputs=[width, height], outputs=[width, height], show_progress=False)
 
-            restore_progress_button.click(
-              fn=restore_progress_call,
-              _js="() => restoreProgress('img2img')",
-              inputs=[],
-              outputs=[
-                  img2img_gallery,
-                  generation_info,
-                  html_info,
-                  html_log,
-              ]
-            )
-
             img2img_interrogate.click(
                 fn=lambda *args: process_interrogate(interrogate, *args),
                 **interrogate_args,
@@ -1574,7 +1547,7 @@ def create_ui():
                 gr.HTML(shared.html("licenses.html"), elem_id="licenses")
 
             gr.Button(value="Show all pages", elem_id="settings_show_all_pages")
-
+            
 
         def unload_sd_weights():
             modules.sd_models.unload_model_weights()

webui.py

@@ -339,7 +339,6 @@ def webui():
         setup_middleware(app)
 
         modules.progress.setup_progress_api(app)
-        modules.progress.setup_current_task_api(app)
 
         if launch_api:
             create_api(app)