Upload llmdolphin.py

llmdolphin.py

@@ -355,6 +355,13 @@ def select_dolphin_language(lang: str):
     return gr.update(value=get_dolphin_sysprompt())
 
 
+@timeout_decorator.timeout(15)
+def get_raw_prompt(msg: str):
+    import re
+    m = re.findall(r'/GENBEGIN/((?:.|\s)+?)/GENEND/', msg)
+    return re.sub(r'[*/:_"#]|\n', ' ', ", ".join(m)).lower() if m else ""
+
+
 @spaces.GPU
 def dolphin_respond(
     message: str,
@@ -430,7 +437,6 @@ def dolphin_respond(
         yield [(outputs, None)]
 
 
-@timeout_decorator.timeout(15, use_signals=False)
 def dolphin_parse(
     history: list[tuple[str, str]],
 ):
@@ -440,8 +446,7 @@ def dolphin_parse(
         msg = history[-1][0]
     except Exception:
         return ""
-    m = re.findall(r'/GENBEGIN/((?:.|\s)+?)/GENEND/', msg)
-    raw_prompt = re.sub(r'[*/:_"#]|\n', ' ', ", ".join(m)).lower() if m else ""
+    raw_prompt = get_raw_prompt(msg)
     prompts = []
     if dolphin_sysprompt_mode == "Japanese to Danbooru Dictionary" and is_japanese(raw_prompt):
         prompts = list_uniq(jatags_to_danbooru_tags(to_list_ja(raw_prompt)) + ["nsfw", "explicit"])
@@ -528,7 +533,6 @@ def dolphin_respond_auto(
         yield [(outputs, None)]
 
 
-@timeout_decorator.timeout(15, use_signals=False)
 def dolphin_parse_simple(
     message: str,
     history: list[tuple[str, str]],
@@ -540,8 +544,7 @@ def dolphin_parse_simple(
         msg = history[-1][0]
     except Exception:
         return ""
-    m = re.findall(r'/GENBEGIN/((?:.|\s)+?)/GENEND/', msg)
-    raw_prompt = re.sub(r'[*/:_"#]|\n', ' ', ", ".join(m)).lower() if m else ""
+    raw_prompt = get_raw_prompt(msg)
     prompts = []
     if dolphin_sysprompt_mode == "Japanese to Danbooru Dictionary" and is_japanese(raw_prompt):
         prompts = list_uniq(jatags_to_danbooru_tags(to_list_ja(raw_prompt)) + ["nsfw", "explicit"])