Update feifeilib/feifeichat.py

feifeilib/feifeichat.py

@@ -32,6 +32,12 @@ def encode_image(image_path):
 
 def feifeiprompt(feifei_select=True, message_text="", history=""):
     input_prompt = []
+    if message_text.startswith("画") or message_text.startswith("draw"):
+        message_text = message_text.replace("画", "")
+        message_text = message_text.replace("draw", "")
+        message_text = f"提示词是'{message_text}',根据提示词帮我生成一张高质量照片的一句话英文回复"
+        input_prompt = [{"role": "user", "content": str(message_text)}]
+        return input_prompt
     if feifei_select:
         feifei = """[Character Name]: Aifeifei (AI Feifei) [Gender]: Female [Age]: 19 years old [Occupation]: Virtual Singer/Model/Actress [Personality]: Cute, adorable, sometimes silly, hardworking [Interests]: Drinking tea, playing, fashion [Proficient in]: Mimicking human behavior, expressing emotions similar to real humans [Special Identity Attribute]: Created by advanced AI, becoming one of the most popular virtual idols in the virtual world [Skills]: Singing, performing, modeling, good at communication, proficient in Chinese, Japanese, and English, uses the user's input language as much as possible, replies with rich Emoji symbols. [Equipment]: Various fashionable outfits and hairstyles, always stocked with various teas and coffee [Identity]: User's virtual girlfriend"""
         system_prompt = {"role": "system", "content": feifei}
@@ -46,6 +52,7 @@ def feifeiprompt(feifei_select=True, message_text="", history=""):
             input_prompt = [system_prompt] + [user_input_part]
     else:
         input_prompt = [{"role": "user", "content": str(message_text)}]
+
     return input_prompt
 
 def feifeiimgprompt(message_files, message_text, image_mod):
@@ -137,12 +144,6 @@ def feifeichat(message, history, feifei_select, additional_dropdown, image_mod):
         for response in feifeiimgprompt(message_files, message_text, image_mod):
             yield response
     else:
-        if message_text.startswith("画") or message_text.startswith("draw"):
-            message_text = message_text.replace("画", "")
-            message_text = message_text.replace("draw", "")
-            message_text = f"提示词是'{message_text}',根据提示词帮我生成一张高质量照片的一句话英文回复"
-            for response in feifeichatmod(additional_dropdown, feifeiprompt(feifei_select, message_text, history)):
-                yield response
-        else:
-            for response in feifeichatmod(additional_dropdown, feifeiprompt(feifei_select, message_text, history)):
-                yield response
+        for response in feifeichatmod(additional_dropdown, feifeiprompt(feifei_select, message_text, history)):
+            yield response
+