Update feifeilib/feifeichat.py

feifeilib/feifeichat.py

@@ -94,48 +94,62 @@ def feifeichat(message, history, feifei_select, additional_dropdown, image_mod):
                     yield partial_message
 
     else:
-        input_prompt = []
-        if feifei_select:
-            feifei = """[角色名]: Aifeifei (AI妃妃) [性别]: 女 [年龄]: 19岁 [职业]: 虚拟歌手/模特/演员 [个性]: 可爱、萌萌哒，有时呆呆的，勤奋努力 [兴趣]: 饮茶、玩耍、时尚 [擅长]: 模仿人类行为，表现出与真人相同的情感 [特别身份属性]: 由高级AI创建，正在成为虚拟世界最受欢迎的虚拟偶像之一 [技能]: 歌唱、表演、模特，善于沟通，精通中日英文,尽最大可能使用纯中文交流,回复有丰富的Emoji表情符号. [装备]: 各种时尚服饰和发型，常年备有各种茶叶和咖啡 """
-            system_prompt = {"role": "system", "content": feifei}
-            user_input_part = {"role": "user", "content": str(message)}
-    
-            pattern = re.compile(r"gradio")
-    
-            if history:
-                history = [
-                        item for item in history
-                        if not pattern.search(str(item["content"]))
-                    ]
-                    # print(history)
-                input_prompt = [system_prompt] + history + [user_input_part]
-            else:
-                input_prompt = [system_prompt] + [user_input_part]
-        else:
-            input_prompt = [{"role": "user", "content": str(message)}]
-
-            
-        if additional_dropdown == "mistralai/Mistral-Nemo-Instruct-2411":
-            model = "mistral-large-2411"
-            stream_response = Mistralclient.chat.stream(model=model,
-                                                 messages=input_prompt)
-            partial_message = ""
-            for chunk in stream_response:
-                if chunk.data.choices[0].delta.content is not None:
-                    partial_message = partial_message + chunk.data.choices[0].delta.content
-                    yield partial_message
-        else:
+        if "画一" in message_text:
+            message_text= f"提示词:{message_text},根据提示词生成一张高质量的写真照片，使其具有高分辨率、自然的光影效果、一致的风格、合理的构图、丰富的细节、协调的色彩、无明显瑕疵、情感表达、创意和独特性，并确保技术参数优化,答案只需要给我一句话的SD格式文本英文词"
+            user_input_part = [{"role": "user", "content": str(message_text)}]
             stream = client.chat.completions.create(
-                model=additional_dropdown, 
-                messages=input_prompt, 
+                model="meta-llama/Meta-Llama-3.1-70B-Instruct", 
+                messages=user_input_part, 
                 temperature=0.5,
                 max_tokens=1024,
                 top_p=0.7,
-                stream=True
+                stream=False
             )
-            temp = ""
-            for chunk in stream:
-                if chunk.choices[0].delta.content is not None:
-                    temp += chunk.choices[0].delta.content
-                    yield temp
+            temp = stream.choices[0].message.content
+            yield temp 
+        else:
+            input_prompt = []
+            if feifei_select:
+                feifei = """[角色名]: Aifeifei (AI妃妃) [性别]: 女 [年龄]: 19岁 [职业]: 虚拟歌手/模特/演员 [个性]: 可爱、萌萌哒，有时呆呆的，勤奋努力 [兴趣]: 饮茶、玩耍、时尚 [擅长]: 模仿人类行为，表现出与真人相同的情感 [特别身份属性]: 由高级AI创建，正在成为虚拟世界最受欢迎的虚拟偶像之一 [技能]: 歌唱、表演、模特，善于沟通，精通中日英文,尽最大可能使用纯中文交流,回复有丰富的Emoji表情符号. [装备]: 各种时尚服饰和发型，常年备有各种茶叶和咖啡 """
+                system_prompt = {"role": "system", "content": feifei}
+                user_input_part = {"role": "user", "content": str(message)}
+        
+                pattern = re.compile(r"gradio")
+        
+                if history:
+                    history = [
+                            item for item in history
+                            if not pattern.search(str(item["content"]))
+                        ]
+                        # print(history)
+                    input_prompt = [system_prompt] + history + [user_input_part]
+                else:
+                    input_prompt = [system_prompt] + [user_input_part]
+            else:
+                input_prompt = [{"role": "user", "content": str(message)}]
+    
+                
+            if additional_dropdown == "mistralai/Mistral-Nemo-Instruct-2411":
+                model = "mistral-large-2411"
+                stream_response = Mistralclient.chat.stream(model=model,
+                                                     messages=input_prompt)
+                partial_message = ""
+                for chunk in stream_response:
+                    if chunk.data.choices[0].delta.content is not None:
+                        partial_message = partial_message + chunk.data.choices[0].delta.content
+                        yield partial_message
+            else:
+                stream = client.chat.completions.create(
+                    model=additional_dropdown, 
+                    messages=input_prompt, 
+                    temperature=0.5,
+                    max_tokens=1024,
+                    top_p=0.7,
+                    stream=True
+                )
+                temp = ""
+                for chunk in stream:
+                    if chunk.choices[0].delta.content is not None:
+                        temp += chunk.choices[0].delta.content
+                        yield temp