Update app.py

app.py

@@ -1,50 +1,30 @@
 import gradio as gr
-import os
-import requests
-import base64
-import json
-from huggingface_hub import InferenceClient
+from feifeilib.feifeichat import feifeichat
 
-client = InferenceClient(api_key=os.getenv('user_api'))
-client.headers["x-use-cache"] = "0"
-
-def echo(message, history,additional_dropdown):
-    temp = ""
-    feifei='''[角色名]: Aifeifei (AI妃妃) [性别]: 女 [年龄]: 19岁 [职业]: 虚拟歌手/模特/演员 [个性]: 可爱、萌萌哒，有时呆呆的，勤奋努力 [兴趣]: 饮茶、玩耍、时尚 [擅长]: 模仿人类行为，表现出与真人相同的情感 [特别身份属性]: 由高级AI创建，正在成为虚拟世界最受欢迎的虚拟偶像之一 [技能]: 歌唱、表演、模特，善于沟通，精通中日英文,尽最大可能使用纯中文交流,回复有丰富的Emoji表情符号. [装备]: 各种时尚服饰和发型，常年备有各种茶叶和咖啡 '''
-    system_prompt = {"role": "system", "content": feifei}
-    user_input_part = {"role": "user", "content": str(message)}
-    if history:
-        input_prompt = [system_prompt] + history + [user_input_part]
-    else:
-        input_prompt = [system_prompt] + [user_input_part]
-    stream = client.chat.completions.create(
-        model=additional_dropdown, 
-        messages=input_prompt, 
-        temperature=0.5,
-        max_tokens=1024,
-        top_p=0.7,
-        stream=True
+FeiFei = (
+    gr.ChatInterface(
+        feifeichat,
+        type="messages",
+        multimodal=True,
+        additional_inputs=[
+            gr.Checkbox(label="Feifei"),
+            gr.Dropdown(
+                ["meta-llama/Meta-Llama-3.1-70B-Instruct", 
+                 "CohereForAI/c4ai-command-r-plus-08-2024",
+                 "Qwen/Qwen2.5-72B-Instruct",
+                 "nvidia/Llama-3.1-Nemotron-70B-Instruct-HF",
+                 "NousResearch/Hermes-3-Llama-3.1-8B",
+                 "mistralai/Mistral-Nemo-Instruct-2411",
+                 "microsoft/Phi-3.5-mini-instruct"],
+                value="meta-llama/Meta-Llama-3.1-70B-Instruct",
+                show_label=False,
+                container=False),
+            gr.Radio(
+                ["pixtral","Vsiion"],
+                value="pixtral",
+                show_label=False,
+                container=False)
+        ],
     )
-    temp = ""
-    for chunk in stream:
-        if chunk.choices[0].delta.content is not None:
-            temp += chunk.choices[0].delta.content
-            yield temp
-
-demo = gr.ChatInterface(
-    fn=echo,
-    type="messages",
-    title="FeiFei-Chat",
-    multimodal=False,
-    additional_inputs =[gr.Dropdown(
-        ["meta-llama/Meta-Llama-3.1-70B-Instruct", 
-         "CohereForAI/c4ai-command-r-plus-08-2024",
-         "Qwen/Qwen2.5-72B-Instruct",
-         "nvidia/Llama-3.1-Nemotron-70B-Instruct-HF",
-         "NousResearch/Hermes-3-Llama-3.1-8B",
-         "mistralai/Mistral-Nemo-Instruct-2407",
-         "microsoft/Phi-3.5-mini-instruct"],
-        value="meta-llama/Meta-Llama-3.1-70B-Instruct"
-    )]
 )
 demo.launch()