+"""
+
+block_css = """
+#notice_markdown {
+ font-size: 110%
+}
+#notice_markdown th {
+ display: none;
+}
+#notice_markdown td {
+ padding-top: 6px;
+ padding-bottom: 6px;
+}
+#model_description_markdown {
+ font-size: 110%
+}
+#leaderboard_markdown {
+ font-size: 110%
+}
+#leaderboard_markdown td {
+ padding-top: 6px;
+ padding-bottom: 6px;
+}
+#leaderboard_dataframe td {
+ line-height: 0.1em;
+}
+#about_markdown {
+ font-size: 110%
+}
+#ack_markdown {
+ font-size: 110%
+}
+#input_box textarea {
+}
+footer {
+ display:none !important
+}
+.image-about img {
+ margin: 0 30px;
+ margin-top: 30px;
+ height: 60px;
+ max-height: 100%;
+ width: auto;
+ float: left;
+.input-image, .image-preview {
+ margin: 0 30px;
+ height: 30px;
+ max-height: 100%;
+ width: auto;
+ max-width: 30%;}
+}
+"""
+
+def enable_buttons_side_by_side():
+ return tuple(gr.update(visible=True, interactive=True) for i in range(6))
+
+def disable_buttons_side_by_side():
+ return tuple(gr.update(visible=i>=4, interactive=False) for i in range(6))
+
+def enable_buttons():
+ return tuple(gr.update(interactive=True) for _ in range(5))
+
+def disable_buttons():
+ return tuple(gr.update(interactive=False) for _ in range(5))
+
+def clear_history():
+ return None, "", None
+
+def clear_history_side_by_side():
+ return None, None, "", None, None
+
+def clear_history_side_by_side_anony():
+ return None, None, "", None, None, gr.Markdown(""), gr.Markdown("")
+
+def clear_history_ie():
+ return None, "", "", "", None, None
+
+def clear_history_side_by_side_ie():
+ return None, None, "", "", "", None, None, None
+
+def clear_history_side_by_side_ie_anony():
+ return None, None, "", "", "", None, None, None, gr.Markdown(""), gr.Markdown("")
+
+def get_ip(request: gr.Request):
+ if request:
+ if "cf-connecting-ip" in request.headers:
+ ip = request.headers["cf-connecting-ip"]
+ else:
+ ip = request.client.host
+ else:
+ ip = None
+ return ip
+
+def get_conv_log_filename():
+ t = datetime.datetime.now()
+ name = os.path.join(LOGDIR, f"{t.year}-{t.month:02d}-{t.day:02d}-conv.json")
+ return name
+
+def save_image_file_on_log_server(image_file:str):
+
+ image_file = Path(image_file).absolute().relative_to(os.getcwd())
+ image_file = str(image_file)
+ # Open the image file in binary mode
+ url = f"{LOG_SERVER_ADDR}/{SAVE_IMAGE}"
+ with open(image_file, 'rb') as f:
+ # Make the POST request, sending the image file and the image path
+ response = requests.post(url, files={'image': f}, data={'image_path': image_file})
+ return response
+
+def append_json_item_on_log_server(json_item: Union[dict, str], log_file: str):
+ if isinstance(json_item, dict):
+ json_item = json.dumps(json_item)
+ log_file = Path(log_file).absolute().relative_to(os.getcwd())
+ log_file = str(log_file)
+ url = f"{LOG_SERVER_ADDR}/{APPEND_JSON}"
+ # Make the POST request, sending the JSON string and the log file name
+ response = requests.post(url, data={'json_str': json_item, 'file_name': log_file})
+ return response
\ No newline at end of file
diff --git a/serve/vote_utils.py b/serve/vote_utils.py
new file mode 100644
index 0000000000000000000000000000000000000000..3c3b03a8d1c677b67ba023275c5a4c4fe719b398
--- /dev/null
+++ b/serve/vote_utils.py
@@ -0,0 +1,657 @@
+import datetime
+import time
+import json
+import uuid
+import gradio as gr
+from pathlib import Path
+from .utils import *
+from .log_utils import build_logger
+from .constants import IMAGE_DIR
+
+ig_logger = build_logger("gradio_web_server_image_generation", "gr_web_image_generation.log") # ig = image generation, loggers for single model direct chat
+igm_logger = build_logger("gradio_web_server_image_generation_multi", "gr_web_image_generation_multi.log") # igm = image generation multi, loggers for side-by-side and battle
+ie_logger = build_logger("gradio_web_server_image_editing", "gr_web_image_editing.log") # ie = image editing, loggers for single model direct chat
+iem_logger = build_logger("gradio_web_server_image_editing_multi", "gr_web_image_editing_multi.log") # iem = image editing multi, loggers for side-by-side and battle
+
+def vote_last_response_ig(state, vote_type, model_selector, request: gr.Request):
+ with open(get_conv_log_filename(), "a") as fout:
+ data = {
+ "tstamp": round(time.time(), 4),
+ "type": vote_type,
+ "model": model_selector,
+ "state": state.dict(),
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+
+def vote_last_response_igm(states, vote_type, model_selectors, request: gr.Request):
+ with open(get_conv_log_filename(), "a") as fout:
+ data = {
+ "tstamp": round(time.time(), 4),
+ "type": vote_type,
+ "models": [x for x in model_selectors],
+ "states": [x.dict() for x in states],
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+ for state in states:
+ output_file = f'{IMAGE_DIR}/generation/{state.conv_id}.jpg'
+ with open(output_file, 'w') as f:
+ state.output.save(f, 'JPEG')
+ save_image_file_on_log_server(output_file)
+
+def vote_last_response_ie(state, vote_type, model_selector, request: gr.Request):
+ with open(get_conv_log_filename(), "a") as fout:
+ data = {
+ "tstamp": round(time.time(), 4),
+ "type": vote_type,
+ "model": model_selector,
+ "state": state.dict(),
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+ output_file = f'{IMAGE_DIR}/edition/{state.conv_id}.jpg'
+ source_file = f'{IMAGE_DIR}/edition/{state.conv_id}_source.jpg'
+ with open(output_file, 'w') as f:
+ state.output.save(f, 'JPEG')
+ with open(source_file, 'w') as sf:
+ state.source_image.save(sf, 'JPEG')
+ save_image_file_on_log_server(output_file)
+ save_image_file_on_log_server(source_file)
+
+def vote_last_response_iem(states, vote_type, model_selectors, request: gr.Request):
+ with open(get_conv_log_filename(), "a") as fout:
+ data = {
+ "tstamp": round(time.time(), 4),
+ "type": vote_type,
+ "models": [x for x in model_selectors],
+ "states": [x.dict() for x in states],
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+ for state in states:
+ output_file = f'{IMAGE_DIR}/edition/{state.conv_id}.jpg'
+ source_file = f'{IMAGE_DIR}/edition/{state.conv_id}_source.jpg'
+ with open(output_file, 'w') as f:
+ state.output.save(f, 'JPEG')
+ with open(source_file, 'w') as sf:
+ state.source_image.save(sf, 'JPEG')
+ save_image_file_on_log_server(output_file)
+ save_image_file_on_log_server(source_file)
+
+
+## Image Generation (IG) Single Model Direct Chat
+def upvote_last_response_ig(state, model_selector, request: gr.Request):
+ ip = get_ip(request)
+ ig_logger.info(f"upvote. ip: {ip}")
+ vote_last_response_ig(state, "upvote", model_selector, request)
+ return ("",) + (disable_btn,) * 3
+
+def downvote_last_response_ig(state, model_selector, request: gr.Request):
+ ip = get_ip(request)
+ ig_logger.info(f"downvote. ip: {ip}")
+ vote_last_response_ig(state, "downvote", model_selector, request)
+ return ("",) + (disable_btn,) * 3
+
+
+def flag_last_response_ig(state, model_selector, request: gr.Request):
+ ip = get_ip(request)
+ ig_logger.info(f"flag. ip: {ip}")
+ vote_last_response_ig(state, "flag", model_selector, request)
+ return ("",) + (disable_btn,) * 3
+
+## Image Generation Multi (IGM) Side-by-Side and Battle
+
+def leftvote_last_response_igm(
+ state0, state1, model_selector0, model_selector1, request: gr.Request
+):
+ igm_logger.info(f"leftvote (named). ip: {get_ip(request)}")
+ vote_last_response_igm(
+ [state0, state1], "leftvote", [model_selector0, model_selector1], request
+ )
+ return ("",) + (disable_btn,) * 4
+
+
+def rightvote_last_response_igm(
+ state0, state1, model_selector0, model_selector1, request: gr.Request
+):
+ igm_logger.info(f"rightvote (named). ip: {get_ip(request)}")
+ vote_last_response_igm(
+ [state0, state1], "rightvote", [model_selector0, model_selector1], request
+ )
+ return ("",) + (disable_btn,) * 4
+
+
+def tievote_last_response_igm(
+ state0, state1, model_selector0, model_selector1, request: gr.Request
+):
+ igm_logger.info(f"tievote (named). ip: {get_ip(request)}")
+ vote_last_response_igm(
+ [state0, state1], "tievote", [model_selector0, model_selector1], request
+ )
+ return ("",) + (disable_btn,) * 4
+
+
+def bothbad_vote_last_response_igm(
+ state0, state1, model_selector0, model_selector1, request: gr.Request
+):
+ igm_logger.info(f"bothbad_vote (named). ip: {get_ip(request)}")
+ vote_last_response_igm(
+ [state0, state1], "bothbad_vote", [model_selector0, model_selector1], request
+ )
+ return ("",) + (disable_btn,) * 4
+
+## Image Editing (IE) Single Model Direct Chat
+
+def upvote_last_response_ie(state, model_selector, request: gr.Request):
+ ip = get_ip(request)
+ ie_logger.info(f"upvote. ip: {ip}")
+ vote_last_response_ie(state, "upvote", model_selector, request)
+ return ("", "", gr.Image(height=512, width=512, type="pil"), "",) + (disable_btn,) * 3
+
+def downvote_last_response_ie(state, model_selector, request: gr.Request):
+ ip = get_ip(request)
+ ie_logger.info(f"downvote. ip: {ip}")
+ vote_last_response_ie(state, "downvote", model_selector, request)
+ return ("", "", gr.Image(height=512, width=512, type="pil"), "",) + (disable_btn,) * 3
+
+def flag_last_response_ie(state, model_selector, request: gr.Request):
+ ip = get_ip(request)
+ ie_logger.info(f"flag. ip: {ip}")
+ vote_last_response_ie(state, "flag", model_selector, request)
+ return ("", "", gr.Image(height=512, width=512, type="pil"), "",) + (disable_btn,) * 3
+
+## Image Editing Multi (IEM) Side-by-Side and Battle
+def leftvote_last_response_iem(
+ state0, state1, model_selector0, model_selector1, request: gr.Request
+):
+ iem_logger.info(f"leftvote (anony). ip: {get_ip(request)}")
+ vote_last_response_iem(
+ [state0, state1], "leftvote", [model_selector0, model_selector1], request
+ )
+ # names = (
+ # "### Model A: " + state0.model_name,
+ # "### Model B: " + state1.model_name,
+ # )
+ names = (state0.model_name, state1.model_name)
+ return names + ("", "", gr.Image(height=512, width=512, type="pil"), "") + (disable_btn,) * 4
+
+def rightvote_last_response_iem(
+ state0, state1, model_selector0, model_selector1, request: gr.Request
+):
+ iem_logger.info(f"rightvote (anony). ip: {get_ip(request)}")
+ vote_last_response_iem(
+ [state0, state1], "rightvote", [model_selector0, model_selector1], request
+ )
+ # names = (
+ # "### Model A: " + state0.model_name,
+ # "### Model B: " + state1.model_name,
+ # )
+ names = (state0.model_name, state1.model_name)
+ return names + ("", "", gr.Image(height=512, width=512, type="pil"), "") + (disable_btn,) * 4
+
+def tievote_last_response_iem(
+ state0, state1, model_selector0, model_selector1, request: gr.Request
+):
+ iem_logger.info(f"tievote (anony). ip: {get_ip(request)}")
+ vote_last_response_iem(
+ [state0, state1], "tievote", [model_selector0, model_selector1], request
+ )
+ names = (
+ "### Model A: " + state0.model_name,
+ "### Model B: " + state1.model_name,
+ )
+ return names + ("", "", gr.Image(height=512, width=512, type="pil"), "") + (disable_btn,) * 4
+
+def bothbad_vote_last_response_iem(
+ state0, state1, model_selector0, model_selector1, request: gr.Request
+):
+ iem_logger.info(f"bothbad_vote (anony). ip: {get_ip(request)}")
+ vote_last_response_iem(
+ [state0, state1], "bothbad_vote", [model_selector0, model_selector1], request
+ )
+ names = (
+ "### Model A: " + state0.model_name,
+ "### Model B: " + state1.model_name,
+ )
+ return names + ("", "", gr.Image(height=512, width=512, type="pil"), "") + (disable_btn,) * 4
+
+
+share_js = """
+function (a, b, c, d) {
+ const captureElement = document.querySelector('#share-region-named');
+ html2canvas(captureElement)
+ .then(canvas => {
+ canvas.style.display = 'none'
+ document.body.appendChild(canvas)
+ return canvas
+ })
+ .then(canvas => {
+ const image = canvas.toDataURL('image/png')
+ const a = document.createElement('a')
+ a.setAttribute('download', 'chatbot-arena.png')
+ a.setAttribute('href', image)
+ a.click()
+ canvas.remove()
+ });
+ return [a, b, c, d];
+}
+"""
+def share_click_igm(state0, state1, model_selector0, model_selector1, request: gr.Request):
+ igm_logger.info(f"share (anony). ip: {get_ip(request)}")
+ if state0 is not None and state1 is not None:
+ vote_last_response_igm(
+ [state0, state1], "share", [model_selector0, model_selector1], request
+ )
+
+def share_click_iem(state0, state1, model_selector0, model_selector1, request: gr.Request):
+ iem_logger.info(f"share (anony). ip: {get_ip(request)}")
+ if state0 is not None and state1 is not None:
+ vote_last_response_iem(
+ [state0, state1], "share", [model_selector0, model_selector1], request
+ )
+
+## All Generation Gradio Interface
+
+class ImageStateIG:
+ def __init__(self, model_name):
+ self.conv_id = uuid.uuid4().hex
+ self.model_name = model_name
+ self.prompt = None
+ self.output = None
+
+ def dict(self):
+ base = {
+ "conv_id": self.conv_id,
+ "model_name": self.model_name,
+ "prompt": self.prompt
+ }
+ return base
+
+class ImageStateIE:
+ def __init__(self, model_name):
+ self.conv_id = uuid.uuid4().hex
+ self.model_name = model_name
+ self.source_prompt = None
+ self.target_prompt = None
+ self.instruct_prompt = None
+ self.source_image = None
+ self.output = None
+
+ def dict(self):
+ base = {
+ "conv_id": self.conv_id,
+ "model_name": self.model_name,
+ "source_prompt": self.source_prompt,
+ "target_prompt": self.target_prompt,
+ "instruct_prompt": self.instruct_prompt
+ }
+ return base
+
+
+
+
+def generate_ig(gen_func, state, text, model_name, request: gr.Request):
+ if not text:
+ raise gr.Error("Prompt cannot be empty.")
+ if not model_name:
+ raise gr.Error("Model name cannot be empty.")
+ if state is None:
+ state = ImageStateIG(model_name)
+ ip = get_ip(request)
+ ig_logger.info(f"generate. ip: {ip}")
+ start_tstamp = time.time()
+ generated_image = gen_func(text, model_name)
+ state.prompt = text
+ state.output = generated_image
+ state.model_name = model_name
+
+ yield state, generated_image
+
+ finish_tstamp = time.time()
+ # logger.info(f"===output===: {output}")
+
+ with open(get_conv_log_filename(), "a") as fout:
+ data = {
+ "tstamp": round(finish_tstamp, 4),
+ "type": "chat",
+ "model": model_name,
+ "gen_params": {},
+ "start": round(start_tstamp, 4),
+ "finish": round(finish_tstamp, 4),
+ "state": state.dict(),
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+
+ output_file = f'{IMAGE_DIR}/generation/{state.conv_id}.jpg'
+ os.makedirs(os.path.dirname(output_file), exist_ok=True)
+ with open(output_file, 'w') as f:
+ state.output.save(f, 'JPEG')
+ save_image_file_on_log_server(output_file)
+
+def generate_igm(gen_func, state0, state1, text, model_name0, model_name1, request: gr.Request):
+ if not text:
+ raise gr.Error("Prompt cannot be empty.")
+ if not model_name0:
+ raise gr.Error("Model name A cannot be empty.")
+ if not model_name1:
+ raise gr.Error("Model name B cannot be empty.")
+ if state0 is None:
+ state0 = ImageStateIG(model_name0)
+ if state1 is None:
+ state1 = ImageStateIG(model_name1)
+ ip = get_ip(request)
+ igm_logger.info(f"generate. ip: {ip}")
+ start_tstamp = time.time()
+ generated_image0, generated_image1 = gen_func(text, model_name0, model_name1)
+ state0.prompt = text
+ state1.prompt = text
+ state0.output = generated_image0
+ state1.output = generated_image1
+ state0.model_name = model_name0
+ state1.model_name = model_name1
+
+ yield state0, state1, generated_image0, generated_image1
+
+ finish_tstamp = time.time()
+ # logger.info(f"===output===: {output}")
+
+ with open(get_conv_log_filename(), "a") as fout:
+ data = {
+ "tstamp": round(finish_tstamp, 4),
+ "type": "chat",
+ "model": model_name0,
+ "gen_params": {},
+ "start": round(start_tstamp, 4),
+ "finish": round(finish_tstamp, 4),
+ "state": state0.dict(),
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+ data = {
+ "tstamp": round(finish_tstamp, 4),
+ "type": "chat",
+ "model": model_name1,
+ "gen_params": {},
+ "start": round(start_tstamp, 4),
+ "finish": round(finish_tstamp, 4),
+ "state": state1.dict(),
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+
+ for i, state in enumerate([state0, state1]):
+ output_file = f'{IMAGE_DIR}/generation/{state.conv_id}.jpg'
+ os.makedirs(os.path.dirname(output_file), exist_ok=True)
+ with open(output_file, 'w') as f:
+ state.output.save(f, 'JPEG')
+ save_image_file_on_log_server(output_file)
+
+def generate_igm_annoy(gen_func, state0, state1, text, model_name0, model_name1, request: gr.Request):
+ if not text:
+ raise gr.Error("Prompt cannot be empty.")
+ if state0 is None:
+ state0 = ImageStateIG(model_name0)
+ if state1 is None:
+ state1 = ImageStateIG(model_name1)
+ ip = get_ip(request)
+ igm_logger.info(f"generate. ip: {ip}")
+ start_tstamp = time.time()
+ generated_image0, generated_image1, model_name0, model_name1 = gen_func(text, model_name0, model_name1)
+ state0.prompt = text
+ state1.prompt = text
+ state0.output = generated_image0
+ state1.output = generated_image1
+ state0.model_name = model_name0
+ state1.model_name = model_name1
+
+ yield state0, state1, generated_image0, generated_image1, \
+ gr.Markdown(f"### Model A: {model_name0}"), gr.Markdown(f"### Model B: {model_name1}")
+
+ finish_tstamp = time.time()
+ # logger.info(f"===output===: {output}")
+
+ with open(get_conv_log_filename(), "a") as fout:
+ data = {
+ "tstamp": round(finish_tstamp, 4),
+ "type": "chat",
+ "model": model_name0,
+ "gen_params": {},
+ "start": round(start_tstamp, 4),
+ "finish": round(finish_tstamp, 4),
+ "state": state0.dict(),
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+ data = {
+ "tstamp": round(finish_tstamp, 4),
+ "type": "chat",
+ "model": model_name1,
+ "gen_params": {},
+ "start": round(start_tstamp, 4),
+ "finish": round(finish_tstamp, 4),
+ "state": state1.dict(),
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+
+ for i, state in enumerate([state0, state1]):
+ output_file = f'{IMAGE_DIR}/generation/{state.conv_id}.jpg'
+ os.makedirs(os.path.dirname(output_file), exist_ok=True)
+ with open(output_file, 'w') as f:
+ state.output.save(f, 'JPEG')
+ save_image_file_on_log_server(output_file)
+
+
+def generate_ie(gen_func, state, source_text, target_text, instruct_text, source_image, model_name, request: gr.Request):
+ if not source_text:
+ raise gr.Error("Source prompt cannot be empty.")
+ if not target_text:
+ raise gr.Error("Target prompt cannot be empty.")
+ if not instruct_text:
+ raise gr.Error("Instruction prompt cannot be empty.")
+ if not source_image:
+ raise gr.Error("Source image cannot be empty.")
+ if not model_name:
+ raise gr.Error("Model name cannot be empty.")
+ if state is None:
+ state = ImageStateIE(model_name)
+ ip = get_ip(request)
+ ig_logger.info(f"generate. ip: {ip}")
+ start_tstamp = time.time()
+ generated_image = gen_func(source_text, target_text, instruct_text, source_image, model_name)
+ state.source_prompt = source_text
+ state.target_prompt = target_text
+ state.instruct_prompt = instruct_text
+ state.source_image = source_image
+ state.output = generated_image
+ state.model_name = model_name
+
+ yield state, generated_image
+
+ finish_tstamp = time.time()
+ # logger.info(f"===output===: {output}")
+
+ with open(get_conv_log_filename(), "a") as fout:
+ data = {
+ "tstamp": round(finish_tstamp, 4),
+ "type": "chat",
+ "model": model_name,
+ "gen_params": {},
+ "start": round(start_tstamp, 4),
+ "finish": round(finish_tstamp, 4),
+ "state": state.dict(),
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+
+ src_img_file = f'{IMAGE_DIR}/edition/{state.conv_id}_src.jpg'
+ os.makedirs(os.path.dirname(src_img_file), exist_ok=True)
+ with open(src_img_file, 'w') as f:
+ state.source_image.save(f, 'JPEG')
+ output_file = f'{IMAGE_DIR}/edition/{state.conv_id}_out.jpg'
+ with open(output_file, 'w') as f:
+ state.output.save(f, 'JPEG')
+ save_image_file_on_log_server(src_img_file)
+ save_image_file_on_log_server(output_file)
+
+def generate_iem(gen_func, state0, state1, source_text, target_text, instruct_text, source_image, model_name0, model_name1, request: gr.Request):
+ if not source_text:
+ raise gr.Error("Source prompt cannot be empty.")
+ if not target_text:
+ raise gr.Error("Target prompt cannot be empty.")
+ if not instruct_text:
+ raise gr.Error("Instruction prompt cannot be empty.")
+ if not source_image:
+ raise gr.Error("Source image cannot be empty.")
+ if not model_name0:
+ raise gr.Error("Model name A cannot be empty.")
+ if not model_name1:
+ raise gr.Error("Model name B cannot be empty.")
+ if state0 is None:
+ state0 = ImageStateIE(model_name0)
+ if state1 is None:
+ state1 = ImageStateIE(model_name1)
+ ip = get_ip(request)
+ igm_logger.info(f"generate. ip: {ip}")
+ start_tstamp = time.time()
+ generated_image0, generated_image1 = gen_func(source_text, target_text, instruct_text, source_image, model_name0, model_name1)
+ state0.source_prompt = source_text
+ state0.target_prompt = target_text
+ state0.instruct_prompt = instruct_text
+ state0.source_image = source_image
+ state0.output = generated_image0
+ state0.model_name = model_name0
+ state1.source_prompt = source_text
+ state1.target_prompt = target_text
+ state1.instruct_prompt = instruct_text
+ state1.source_image = source_image
+ state1.output = generated_image1
+ state1.model_name = model_name1
+
+ yield state0, state1, generated_image0, generated_image1
+
+ finish_tstamp = time.time()
+ # logger.info(f"===output===: {output}")
+
+ with open(get_conv_log_filename(), "a") as fout:
+ data = {
+ "tstamp": round(finish_tstamp, 4),
+ "type": "chat",
+ "model": model_name0,
+ "gen_params": {},
+ "start": round(start_tstamp, 4),
+ "finish": round(finish_tstamp, 4),
+ "state": state0.dict(),
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+ data = {
+ "tstamp": round(finish_tstamp, 4),
+ "type": "chat",
+ "model": model_name1,
+ "gen_params": {},
+ "start": round(start_tstamp, 4),
+ "finish": round(finish_tstamp, 4),
+ "state": state1.dict(),
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+
+ for i, state in enumerate([state0, state1]):
+ src_img_file = f'{IMAGE_DIR}/edition/{state.conv_id}_src.jpg'
+ os.makedirs(os.path.dirname(src_img_file), exist_ok=True)
+ with open(src_img_file, 'w') as f:
+ state.source_image.save(f, 'JPEG')
+ output_file = f'{IMAGE_DIR}/edition/{state.conv_id}_out.jpg'
+ with open(output_file, 'w') as f:
+ state.output.save(f, 'JPEG')
+ save_image_file_on_log_server(src_img_file)
+ save_image_file_on_log_server(output_file)
+
+
+def generate_iem_annoy(gen_func, state0, state1, source_text, target_text, instruct_text, source_image, model_name0, model_name1, request: gr.Request):
+ if not source_text:
+ raise gr.Error("Source prompt cannot be empty.")
+ if not target_text:
+ raise gr.Error("Target prompt cannot be empty.")
+ if not instruct_text:
+ raise gr.Error("Instruction prompt cannot be empty.")
+ if not source_image:
+ raise gr.Error("Source image cannot be empty.")
+ if state0 is None:
+ state0 = ImageStateIE(model_name0)
+ if state1 is None:
+ state1 = ImageStateIE(model_name1)
+ ip = get_ip(request)
+ igm_logger.info(f"generate. ip: {ip}")
+ start_tstamp = time.time()
+ generated_image0, generated_image1, model_name0, model_name1 = gen_func(source_text, target_text, instruct_text, source_image, model_name0, model_name1)
+ state0.source_prompt = source_text
+ state0.target_prompt = target_text
+ state0.instruct_prompt = instruct_text
+ state0.source_image = source_image
+ state0.output = generated_image0
+ state0.model_name = model_name0
+ state1.source_prompt = source_text
+ state1.target_prompt = target_text
+ state1.instruct_prompt = instruct_text
+ state1.source_image = source_image
+ state1.output = generated_image1
+ state1.model_name = model_name1
+
+ yield state0, state1, generated_image0, generated_image1, \
+ gr.Markdown(f"### Model A: {model_name0}"), gr.Markdown(f"### Model B: {model_name1}")
+
+ finish_tstamp = time.time()
+ # logger.info(f"===output===: {output}")
+
+ with open(get_conv_log_filename(), "a") as fout:
+ data = {
+ "tstamp": round(finish_tstamp, 4),
+ "type": "chat",
+ "model": model_name0,
+ "gen_params": {},
+ "start": round(start_tstamp, 4),
+ "finish": round(finish_tstamp, 4),
+ "state": state0.dict(),
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+ data = {
+ "tstamp": round(finish_tstamp, 4),
+ "type": "chat",
+ "model": model_name1,
+ "gen_params": {},
+ "start": round(start_tstamp, 4),
+ "finish": round(finish_tstamp, 4),
+ "state": state1.dict(),
+ "ip": get_ip(request),
+ }
+ fout.write(json.dumps(data) + "\n")
+ append_json_item_on_log_server(data, get_conv_log_filename())
+
+ for i, state in enumerate([state0, state1]):
+ src_img_file = f'{IMAGE_DIR}/edition/{state.conv_id}_src.jpg'
+ os.makedirs(os.path.dirname(src_img_file), exist_ok=True)
+ with open(src_img_file, 'w') as f:
+ state.source_image.save(f, 'JPEG')
+ output_file = f'{IMAGE_DIR}/edition/{state.conv_id}_out.jpg'
+ with open(output_file, 'w') as f:
+ state.output.save(f, 'JPEG')
+ save_image_file_on_log_server(src_img_file)
+ save_image_file_on_log_server(output_file)
\ No newline at end of file
diff --git a/style.css b/style.css
deleted file mode 100644
index 114adf441e9032febb46bc056b2a8bb651075f0d..0000000000000000000000000000000000000000
--- a/style.css
+++ /dev/null
@@ -1,28 +0,0 @@
-body {
- padding: 2rem;
- font-family: -apple-system, BlinkMacSystemFont, "Arial", sans-serif;
-}
-
-h1 {
- font-size: 16px;
- margin-top: 0;
-}
-
-p {
- color: rgb(107, 114, 128);
- font-size: 15px;
- margin-bottom: 10px;
- margin-top: 5px;
-}
-
-.card {
- max-width: 620px;
- margin: 0 auto;
- padding: 16px;
- border: 1px solid lightgray;
- border-radius: 16px;
-}
-
-.card p:last-child {
- margin-bottom: 0;
-}