Init StarrySpace,Add image(s) using LFS

app.py

@@ -1,64 +1,185 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
+import openai
+import os
+import time
 
+# --- 配置 ---
+ARK_API_KEY = os.environ.get("ARK_API_KEY")
+if not ARK_API_KEY:
+    raise ValueError("ARK_API_KEY not found. Please set it in Space Secrets.")
+
+client = openai.OpenAI(api_key=ARK_API_KEY, base_url="https://ark.cn-beijing.volces.com/api/v3")
+
+# --- 可选模型列表 ---
+# 键是用户看到的显示名称，值是实际的API模型ID
+AVAILABLE_MODELS = {
+    "DeepSeek R1 (250120)": "deepseek-r1-250120",
+    "DeepSeek v3 (250324)": "deepseek-v3-250324" # 示例，你需要有这个模型的权限
+    # 你可以添加更多你账户下可用的模型
+    # "模型名称A": "模型ID_A",
+    # "模型名称B": "模型ID_B",
+}
+DEFAULT_MODEL_DISPLAY_NAME = list(AVAILABLE_MODELS.keys())[0] # 默认选择第一个模型
+
+# --- 自定义 CSS ---
+custom_css = """
+body {
+    background-color: #f0f2f5;
+    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+}
+.gradio-container {
+    border-radius: 15px !important;
+    box-shadow: 0 4px 12px rgba(0,0,0,0.1) !important;
+    background-color: #ffffff;
+}
+.message-bubble-user {
+    background-color: #007bff !important;
+    color: white !important;
+    border-radius: 15px 15px 0 15px !important;
+    align-self: flex-end;
+    max-width: 70%;
+}
+.message-bubble-bot {
+    background-color: #e9ecef !important;
+    color: #333 !important;
+    border-radius: 15px 15px 15px 0 !important;
+    align-self: flex-start;
+    max-width: 70%;
+}
+.message-bubble-user .message-text, .message-bubble-bot .message-text {
+    font-size: 1rem !important;
+    padding: 10px 15px !important;
+}
+.avatar-container img {
+    border-radius: 50% !important;
+    width: 40px !important;
+    height: 40px !important;
+    object-fit: cover;
+    margin: 5px !important;
+    box-shadow: 0 1px 3px rgba(0,0,0,0.1);
+}
+.textbox textarea {
+    border-radius: 20px !important;
+    border: 1px solid #ced4da !important;
+    padding: 10px 15px !important;
+    font-size: 1rem !important;
+    box-shadow: none !important;
+}
+.submit-button {
+    background-color: #007bff !important;
+    color: white !important;
+    border-radius: 20px !important;
+    font-weight: bold !important;
+    padding: 10px 20px !important;
+    transition: background-color 0.2s ease-in-out;
+}
+.submit-button:hover {
+    background-color: #0056b3 !important;
+}
+.gr-prose h1, .gr-prose h2, .gr-prose h3 {
+    color: #343a40;
+    text-align: center;
+}
+.gr-prose p {
+    color: #495057;
+    text-align: center;
+    font-size: 1.1em;
+}
+/* 为下拉菜单添加一些边距 */
+.gradio-dropdown {
+    margin-bottom: 15px !important;
+}
 """
-For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-"""
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
+# --- 聊天逻辑 (修改以接收模型ID) ---
+def call_openai_with_history(message: str, history: list, selected_model_display_name: str):
+    """
+    调用 OpenAI API，并处理聊天历史。
+    message: 当前用户输入
+    history: Gradio 的聊天历史
+    selected_model_display_name: 用户从下拉菜单选择的模型显示名称
+    """
+    actual_model_id = AVAILABLE_MODELS.get(selected_model_display_name, AVAILABLE_MODELS[DEFAULT_MODEL_DISPLAY_NAME])
+    print(f"使用模型: {actual_model_id}") # 调试信息
 
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
-    messages = [{"role": "system", "content": system_message}]
+    messages_for_api = [{"role": "system", "content": "你是一个乐于助人、风趣幽默的AI助手。"}]
 
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
+    for user_msg, bot_msg in history:
+        # Gradio 的 history 格式是 [[user_msg1, bot_msg1], ...]
+        # 当 chatbot type='messages' 时，history 格式是 [{'role':'user', 'content':...}, {'role':'assistant', 'content':...}]
+        # 这里需要根据 chatbot 的 type 来调整 history 的处理方式
+        # 由于 chatbot_component.type = 'messages', history 已经是 OpenAI 期望的格式了
+        # 但是 ChatInterface 传递给 fn 的 history 仍然是 [[user, bot], ...] 格式
+        # 所以我们仍然需要转换
+        if isinstance(user_msg, str): # 确保是 [[user, bot], ...] 格式
+            messages_for_api.append({"role": "user", "content": user_msg})
+        if bot_msg and isinstance(bot_msg, str):
+             messages_for_api.append({"role": "assistant", "content": bot_msg})
 
-    messages.append({"role": "user", "content": message})
+    messages_for_api.append({"role": "user", "content": message})
 
-    response = ""
+    try:
+        response = client.chat.completions.create(
+            model=actual_model_id, # 使用选择的模型ID
+            messages=messages_for_api,
+            temperature=0.7,
+            max_tokens=1024, # 可以适当增加 max_tokens
+        )
+        bot_response = response.choices[0].message.content
+        return bot_response
 
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content
+    except openai.APIError as e:
+        error_message = f"OpenAI API Error: {e}"
+        print(error_message)
+        if hasattr(e, 'message'):
+            return f"抱歉，调用AI服务时出错: {e.message}"
+        return f"抱歉，调用AI服务时出错: {str(e)}"
+    except Exception as e:
+        error_message = f"An unexpected error occurred: {e}"
+        print(error_message)
+        return "抱歉，发生了一个未知错误。"
 
-        response += token
-        yield response
 
+# --- Gradio 界面 ---
+with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
+    gr.Markdown("# 🤖 Starry的智能AI聊天伙伴")
+    gr.Markdown("与基于大模型的AI助手进行友好对话吧！")
 
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (nucleus sampling)",
-        ),
-    ],
-)
+    # 添加模型选择下拉菜单
+    model_selector = gr.Dropdown(
+        choices=list(AVAILABLE_MODELS.keys()), # 用户看到的选项是显示名称
+        value=DEFAULT_MODEL_DISPLAY_NAME,     # 默认值
+        label="选择AI模型",
+        info="更改模型后，新的对话将使用所选模型。",
+        elem_classes="gradio-dropdown" # 应用自定义CSS类 (可选)
+    )
+
+    chatbot_component = gr.Chatbot(
+        label="聊天窗口",
+        type='messages', # 确保使用 'messages' 类型，以便更好地处理头像和角色
+        avatar_images=("images/user_avatar.png", "images/bot_avatar.png"),
+        height=500,
+        show_label=False # 隐藏 Chatbot 自己的标签，因为上面已经有标题了
+    )
+
+    chat_interface = gr.ChatInterface(
+        fn=call_openai_with_history,
+        chatbot=chatbot_component,
+        additional_inputs=[model_selector], # 将下拉菜单作为额外输入传递给 fn
+        examples=[
+            ["你好呀！"],
+            ["给我讲个关于太空的笑话。"],
+            ["用Python写一个快速排序算法。"],
+        ],
+        submit_btn="➤ 发送",
+    )
 
+    # 如果你的 Gradio 版本较低，可能没有 avatar_images 直接在 Chatbot 构造函数里
+    # 如果头像图片不显示，请确保 `images` 文件夹和里面的 `user_avatar.png` `bot_avatar.png`
+    # 文件与你的 app.py 在同一级别，或者你提供了正确的相对/绝对路径。
+    # 对于 Hugging Face Spaces，通常将 `images` 文件夹放在仓库根目录。
 
+# --- 启动应用 ---
 if __name__ == "__main__":
-    demo.launch()
+    # demo.queue() # 如果 API 调用时间较长，可以启用队列
+    demo.launch()