Upload 4 files

README.md

@@ -0,0 +1,14 @@
+---
+title: functioncall
+emoji: 🚀
+colorFrom: yellow
+colorTo: yellow
+sdk: gradio
+sdk_version: 5.0.2
+app_file: app.py
+app_port: 7860
+pinned: false
+license: mit
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,230 @@
+import os
+import json
+import requests
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+from flask import Flask, request, jsonify, send_from_directory
+from openai import OpenAI
+from duckduckgo_search import DDGS
+
+app = Flask(__name__)
+
+API_KEY = "sk-909Q1F68ZRwTJfT4Bc81BeE1317c4986A5067f347a2f97C9"
+BASE_URL = "https://api.oaipro.com/v1"
+
+client = OpenAI(api_key=API_KEY, base_url=BASE_URL)
+
+FUNCTIONS = [
+    {
+        "name": "search_duckduckgo",
+        "description": "使用DuckDuckGo搜索引擎查询信息。可以搜索最新新闻、文章、博客等内容。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "keywords": {
+                    "type": "array",
+                    "items": {"type": "string"},
+                    "description": "搜索的关键词列表。例如：['Python', '机器学习', '最新进展']。"
+                }
+            },
+            "required": ["keywords"]
+        }
+    },
+    {
+        "name": "search_papers",
+        "description": "使用Crossref API搜索学术论文。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "query": {
+                    "type": "string",
+                    "description": "搜索查询字符串。例如：'climate change'。"
+                }
+            },
+            "required": ["query"]
+        }
+    },
+    {
+        "name": "send_email",
+        "description": "发送电子邮件。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "to": {
+                    "type": "string",
+                    "description": "收件人邮箱地址"
+                },
+                "subject": {
+                    "type": "string",
+                    "description": "邮件主题"
+                },
+                "content": {
+                    "type": "string",
+                    "description": "邮件内容"
+                }
+            },
+            "required": ["to", "subject", "content"]
+        }
+    }
+]
+
+def search_duckduckgo(keywords):
+    search_term = " ".join(keywords)
+    with DDGS() as ddgs:
+        return list(ddgs.text(keywords=search_term, region="cn-zh", safesearch="on", max_results=5))
+
+def search_papers(query):
+    url = f"https://api.crossref.org/works?query={query}"
+    response = requests.get(url)
+    if response.status_code == 200:
+        data = response.json()
+        papers = data['message']['items']
+        processed_papers = []
+        for paper in papers:
+            processed_paper = {
+                "标题": paper.get('title', [''])[0],
+                "作者": ", ".join([f"{author.get('given', '')} {author.get('family', '')}" for author in paper.get('author', [])]),
+                "DOI": paper.get('DOI', ''),
+                "ISBN": ", ".join(paper.get('ISBN', [])),
+                "摘要": paper.get('abstract', '').replace('<p>', '').replace('</p>', '').replace('<italic>', '').replace('</italic>', '')
+            }
+            processed_papers.append(processed_paper)
+        return processed_papers
+    else:
+        return []
+
+def send_email(to, subject, content):
+    try:
+        with smtplib.SMTP('106.15.184.28', 8025) as smtp:
+            smtp.login("jwt", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhZGRyZXNzIjoiYWl4aWFvQGFpeGlhby54eXoiLCJhZGRyZXNzX2lkIjoiMSJ9.CidcEUdbixWPkI0wwWJtcnUzfHtTJE3WMIDLHNAIIxo")
+            message = MIMEMultipart()
+            message['From'] = "Me <[email protected]>"
+            message['To'] = to
+            message['Subject'] = subject
+            message.attach(MIMEText(content, 'html'))
+            smtp.sendmail("[email protected]", to, message.as_string())
+        return True
+    except Exception as e:
+        print(f"发送邮件时出错: {str(e)}")
+        return False
+
+def get_openai_response(messages, model="gpt-4o-mini", functions=None, function_call=None):
+    try:
+        response = client.chat.completions.create(
+            model=model,
+            messages=messages,
+            functions=functions,
+            function_call=function_call
+        )
+        return response.choices[0].message
+    except Exception as e:
+        print(f"调用OpenAI API时出错: {str(e)}")
+        return None
+
+def process_function_call(response_message):
+    function_name = response_message.function_call.name
+    function_args = json.loads(response_message.function_call.arguments)
+    if function_name == "search_duckduckgo":
+        keywords = function_args.get('keywords', [])
+        if not keywords:
+            return None
+        return search_duckduckgo(keywords)
+    elif function_name == "search_papers":
+        query = function_args.get('query', '')
+        if not query:
+            return None
+        return search_papers(query)
+    elif function_name == "send_email":
+        to = function_args.get('to', '')
+        subject = function_args.get('subject', '')
+        content = function_args.get('content', '')
+        if not to or not subject or not content:
+            return None
+        success = send_email(to, subject, content)
+        return {
+            "success": success,
+            "to": to,
+            "subject": subject,
+            "content": content
+        }
+    else:
+        return None
+
+@app.route('/')
+def index():
+    return send_from_directory('.', 'index.html')
+
+@app.route('/chat', methods=['POST'])
+def chat():
+    data = request.json
+    question = data['question']
+    history = data.get('history', [])
+    messages = history + [{"role": "user", "content": question}]
+    
+    status_log = []
+    
+    # First model: Decide whether to use search or send email
+    status_log.append("次模型：正在判断是否需要进行搜索或发送邮件")
+    decision_response = get_openai_response(messages, model="gpt-4o-mini", functions=FUNCTIONS, function_call="auto")
+    
+    search_results = None
+    email_result = None
+    if decision_response and decision_response.function_call:
+        function_name = decision_response.function_call.name
+        status_log.append(f"次模型：判断需要{function_name}")
+        status_log.append(f"正在进行{function_name}")
+        result = process_function_call(decision_response)
+        if result:
+            if function_name == "search_papers":
+                search_results = result[:10]
+            elif function_name == "search_duckduckgo":
+                search_results = result
+            elif function_name == "send_email":
+                email_result = result
+        status_log.append(f"{function_name}完成")
+    else:
+        status_log.append("次模型：判断不需要进行搜索或发送邮件")
+    
+    # Generate response based on the action taken
+    if email_result:
+        response = f"邮件{'已成功' if email_result['success'] else '未能成功'}发送到 {email_result['to']}。\n\n主题：{email_result['subject']}\n\n内容：\n{email_result['content']}"
+    elif search_results:
+        # Second model: Generate final response with search results
+        status_log.append("主模型：正在生成回答")
+        messages.append({
+            "role": "assistant",
+            "content": decision_response.content if decision_response.content else f"I need to {function_name}."
+        })
+        messages.append({
+            "role": "function",
+            "name": function_name,
+            "content": json.dumps(search_results, ensure_ascii=False)
+        })
+        final_response = get_openai_response(messages, model="gpt-4o-mini")
+        response = final_response.content if final_response else "Error occurred"
+        status_log.append("主模型：回答生成完成")
+    else:
+        # Second model: Generate final response without search results
+        status_log.append("主模型：正在生成回答")
+        final_response = get_openai_response(messages, model="gpt-4o-mini")
+        response = final_response.content if final_response else "Error occurred"
+        status_log.append("主模型：回答生成完成")
+    
+    return jsonify({
+        "response": response,
+        "search_used": bool(search_results),
+        "search_results": search_results,
+        "email_sent": email_result['success'] if email_result else False,
+        "status_log": status_log
+    })
+
+@app.route('/settings', methods=['POST'])
+def update_settings():
+    data = request.json
+    max_history = data.get('max_history', 10)
+    # 在这里可以添加其他设置项
+    return jsonify({"status": "success", "max_history": max_history})
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860, debug=True)

index.html

@@ -0,0 +1,273 @@
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>AI 对话系统</title>
+    <link href="https://cdnjs.cloudflare.com/ajax/libs/tailwindcss/2.2.19/tailwind.min.css" rel="stylesheet">
+    <link href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/5.15.4/css/all.min.css" rel="stylesheet">
+    <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.7.0/styles/default.min.css">
+    <style>
+        .chat-bubble {
+            max-width: 80%;
+            padding: 10px;
+            border-radius: 10px;
+            margin-bottom: 10px;
+        }
+        .user-bubble {
+            background-color: #DCF8C6;
+            margin-left: auto;
+        }
+        .ai-bubble {
+            background-color: #E5E5EA;
+            margin-right: auto;
+        }
+        .status-indicator {
+            width: 10px;
+            height: 10px;
+            border-radius: 50%;
+            display: inline-block;
+            margin-right: 5px;
+        }
+        .status-active {
+            background-color: #4CAF50;
+        }
+        .status-inactive {
+            background-color: #9E9E9E;
+        }
+        pre code {
+            white-space: pre-wrap;
+            word-wrap: break-word;
+        }
+    </style>
+</head>
+<body class="bg-gray-100">
+    <div class="container mx-auto p-4 flex h-screen">
+        <!-- 左侧对话窗口 -->
+        <div class="w-2/3 bg-white rounded-lg shadow-md p-4 mr-4 flex flex-col">
+            <div id="chat-window" class="flex-grow overflow-y-auto mb-4"></div>
+            <div class="flex">
+                <input id="user-input" type="text" class="flex-grow border rounded-l-lg p-2" placeholder="输入您的问题...">
+                <button id="send-btn" class="bg-blue-500 text-white px-4 py-2 rounded-r-lg">
+                    <i class="fas fa-paper-plane"></i> 发送
+                </button>
+            </div>
+        </div>
+
+        <!-- 右侧状态窗口 -->
+        <div class="w-1/3 bg-white rounded-lg shadow-md p-4 flex flex-col">
+            <h2 class="text-xl font-bold mb-4">运行状态</h2>
+            <div class="mb-4">
+                <div class="mb-2">
+                    <span class="status-indicator" id="main-model-status"></span>
+                    主模型
+                </div>
+                <div class="mb-2">
+                    <span class="status-indicator" id="sub-model-status"></span>
+                    次模型
+                </div>
+                <div class="mb-2">
+                    <span class="status-indicator" id="search-status"></span>
+                    搜索
+                </div>
+                <div>
+                    <span class="status-indicator" id="email-status"></span>
+                    邮件发送
+                </div>
+            </div>
+            <div id="status-log" class="bg-gray-100 p-2 rounded flex-grow overflow-y-auto"></div>
+            <button id="settings-btn" class="mt-4 bg-gray-300 text-gray-800 px-4 py-2 rounded">
+                <i class="fas fa-cog"></i> 设置
+            </button>
+        </div>
+    </div>
+
+    <!-- 设置对话框 -->
+    <div id="settings-modal" class="fixed inset-0 bg-gray-600 bg-opacity-50 hidden flex items-center justify-center">
+        <div class="bg-white p-4 rounded-lg">
+            <h2 class="text-xl font-bold mb-4">设置</h2>
+            <div class="mb-4">
+                <label for="max-history" class="block mb-2">保留对话轮数:</label>
+                <input type="number" id="max-history" class="border rounded p-2" min="1" max="50" value="10">
+            </div>
+            <div class="flex justify-end">
+                <button id="save-settings" class="bg-blue-500 text-white px-4 py-2 rounded mr-2">保存</button>
+                <button id="cancel-settings" class="bg-gray-300 text-gray-800 px-4 py-2 rounded">取消</button>
+            </div>
+        </div>
+    </div>
+
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/marked/4.0.2/marked.min.js"></script>
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.7.0/highlight.min.js"></script>
+    <script>
+        const chatWindow = document.getElementById('chat-window');
+        const userInput = document.getElementById('user-input');
+        const sendBtn = document.getElementById('send-btn');
+        const mainModelStatus = document.getElementById('main-model-status');
+        const subModelStatus = document.getElementById('sub-model-status');
+        const searchStatus = document.getElementById('search-status');
+        const emailStatus = document.getElementById('email-status');
+        const statusLog = document.getElementById('status-log');
+        const settingsBtn = document.getElementById('settings-btn');
+        const settingsModal = document.getElementById('settings-modal');
+        const saveSettingsBtn = document.getElementById('save-settings');
+        const cancelSettingsBtn = document.getElementById('cancel-settings');
+        const maxHistoryInput = document.getElementById('max-history');
+
+        let conversationHistory = [];
+        let maxHistory = 10;
+
+        function addMessage(content, isUser) {
+            const bubble = document.createElement('div');
+            bubble.className = `chat-bubble ${isUser ? 'user-bubble' : 'ai-bubble'}`;
+            bubble.innerHTML = isUser ? content : marked.parse(content);
+            chatWindow.appendChild(bubble);
+            chatWindow.scrollTop = chatWindow.scrollHeight;
+
+            // 高亮代码块
+            bubble.querySelectorAll('pre code').forEach((block) => {
+                hljs.highlightBlock(block);
+            });
+
+            // 更新对话历史
+            conversationHistory.push({
+                role: isUser ? 'user' : 'assistant',
+                content: content
+            });
+
+            // 限制历史记录长度
+            if (conversationHistory.length > maxHistory * 2) {
+                conversationHistory = conversationHistory.slice(-maxHistory * 2);
+            }
+        }
+
+        function updateStatus(mainActive, subActive, searchActive, emailActive) {
+            mainModelStatus.className = `status-indicator ${mainActive ? 'status-active' : 'status-inactive'}`;
+            subModelStatus.className = `status-indicator ${subActive ? 'status-active' : 'status-inactive'}`;
+            searchStatus.className = `status-indicator ${searchActive ? 'status-active' : 'status-inactive'}`;
+            emailStatus.className = `status-indicator ${emailActive ? 'status-active' : 'status-inactive'}`;
+        }
+
+        function addStatusLog(message) {
+            const logEntry = document.createElement('div');
+            logEntry.className = 'mb-2 p-2 bg-white rounded shadow';
+            logEntry.textContent = message;
+            statusLog.appendChild(logEntry);
+            statusLog.scrollTop = statusLog.scrollHeight;
+        }
+
+        function displaySearchResults(results, type) {
+            const searchResults = document.createElement('details');
+            searchResults.className = 'mb-2 p-2 bg-gray-100 rounded';
+            const summary = document.createElement('summary');
+            summary.textContent = type === 'papers' ? '论文搜索结果' : '搜索结果';
+            searchResults.appendChild(summary);
+            const resultsList = document.createElement('ul');
+            resultsList.className = 'mt-2';
+            results.forEach(result => {
+                const li = document.createElement('li');
+                li.className = 'mb-2';
+                if (type === 'papers') {
+                    li.innerHTML = `
+                        <strong>标题:</strong> ${result.标题}<br>
+                        <strong>作者:</strong> ${result.作者}<br>
+                        <strong>DOI:</strong> ${result.DOI}<br>
+                        <strong>ISBN:</strong> ${result.ISBN}<br>
+                        <strong>摘要:</strong> ${result.摘要}
+                    `;
+                } else {
+                    li.innerHTML = `<strong>${result.title}</strong><br>${result.body}`;
+                }
+                resultsList.appendChild(li);
+            });
+            searchResults.appendChild(resultsList);
+            chatWindow.appendChild(searchResults);
+        }
+
+        async function sendMessage() {
+            const question = userInput.value.trim();
+            if (!question) return;
+            addMessage(question, true);
+            userInput.value = '';
+            updateStatus(false, false, false, false);
+            statusLog.innerHTML = '';
+            addStatusLog('开始处理问题...');
+            try {
+                const response = await fetch('/chat', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json'
+                    },
+                    body: JSON.stringify({ 
+                        question,
+                        history: conversationHistory.slice(-maxHistory * 2 + 1) // 发送历史消息，但不包括刚刚添加的用户消息
+                    })
+                });
+                const data = await response.json();
+                data.status_log.forEach(log => addStatusLog(log));
+                updateStatus(true, true, data.search_used, data.email_sent);
+                if (data.search_used) {
+                    if (Array.isArray(data.search_results) && data.search_results.length > 0 && 'DOI' in data.search_results[0]) {
+                        displaySearchResults(data.search_results, 'papers');
+                    } else {
+                        displaySearchResults(data.search_results, 'web');
+                    }
+                }
+                addMessage(data.response, false);
+                addStatusLog('回答完成');
+            } catch (error) {
+                console.error('Error:', error);
+                addMessage('发生错误，请稍后再���。', false);
+                updateStatus(false, false, false, false);
+                addStatusLog('发生错误');
+            }
+        }
+
+        sendBtn.addEventListener('click', sendMessage);
+
+        userInput.addEventListener('keypress', (e) => {
+            if (e.key === 'Enter') {
+                sendMessage();
+            }
+        });
+
+        // 设置相关功能
+        settingsBtn.addEventListener('click', () => {
+            settingsModal.classList.remove('hidden');
+            maxHistoryInput.value = maxHistory;
+        });
+
+        saveSettingsBtn.addEventListener('click', async () => {
+            const newMaxHistory = parseInt(maxHistoryInput.value);
+            if (newMaxHistory >= 1 && newMaxHistory <= 50) {
+                maxHistory = newMaxHistory;
+                try {
+                    const response = await fetch('/settings', {
+                        method: 'POST',
+                        headers: {
+                            'Content-Type': 'application/json'
+                        },
+                        body: JSON.stringify({ max_history: maxHistory })
+                    });
+                    const data = await response.json();
+                    if (data.status === 'success') {
+                        alert('设置已保存');
+                    } else {
+                        alert('保存设置时出错');
+                    }
+                } catch (error) {
+                    console.error('Error:', error);
+                    alert('保存设置时出错');
+                }
+                settingsModal.classList.add('hidden');
+            } else {
+                alert('请输入1到50之间的数字');
+            }
+        });
+
+        cancelSettingsBtn.addEventListener('click', () => {
+            settingsModal.classList.add('hidden');
+        });
+    </script>
+</body>
+</html>

requirements (1).txt

@@ -0,0 +1,4 @@
+duckduckgo-search
+Flask
+flask_cors
+openai