Spaces:
Running
Running
| from fastapi import FastAPI, Form | |
| from fastapi.responses import HTMLResponse | |
| from jinja2 import Template | |
| import markdown | |
| import time | |
| from datetime import datetime, timedelta | |
| from apscheduler.schedulers.background import BackgroundScheduler | |
| from agents import DeepResearchAgent, get_llms | |
| import threading | |
| from queue import Queue | |
| lock = threading.Lock() | |
| app = FastAPI() | |
| # 每日最大回复次数 | |
| MAX_REPLIES_PER_DAY = 100 | |
| # 当日回复次数计数器 | |
| reply_count = 0 | |
| # 启动时设置计数器重置 | |
| last_reset_time = datetime.now() | |
| # HTML模板 | |
| html_template = """ | |
| <!DOCTYPE html> | |
| <html> | |
| <head> | |
| <title>CoI Agent online demo 😊</title> | |
| <style> | |
| body { | |
| font-family: 'Arial', sans-serif; | |
| background-color: #f4f4f9; | |
| margin: 0; | |
| padding: 0; | |
| display: flex; | |
| justify-content: center; | |
| align-items: center; | |
| min-height: 100vh; | |
| } | |
| .container { | |
| width: 95%; | |
| max-width: 1200px; | |
| background-color: #fff; | |
| padding: 2rem; | |
| border-radius: 10px; | |
| box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1); | |
| } | |
| h1 { | |
| font-size: 2rem; | |
| margin-bottom: 1.5rem; | |
| color: #333; | |
| text-align: center; | |
| } | |
| form { | |
| margin-bottom: 1.5rem; | |
| } | |
| .form-group { | |
| display: flex; | |
| justify-content: space-between; | |
| align-items: center; | |
| margin-bottom: 1.5rem; | |
| } | |
| .form-group label { | |
| flex: 0; | |
| font-size: 1 rem; /* 增大字体 */ | |
| color: #333; | |
| margin-right: 0.5rem; | |
| background-color: #f0f8ff; /* 气泡背景颜色 */ | |
| padding: 0.5rem 1rem; /* 气泡内边距 */ | |
| border-radius: 10px; /* 气泡圆角 */ | |
| text-shadow: 1px 1px 2px rgba(0, 0, 0, 0.1); /* 艺术字效果 */ | |
| font-family: 'Times new roman', cursive, sans-serif; /* 艺术字体 */ | |
| box-shadow: 0 0 5px rgba(0, 0, 0, 0.1); /* 气泡阴影 */ | |
| } | |
| .form-group input { | |
| flex: 4; | |
| padding: 0.6rem; | |
| font-size: 1rem; | |
| border: 1px solid #ccc; | |
| border-radius: 5px; | |
| margin-left: 1rem; | |
| } | |
| .form-group button { | |
| flex: 0; | |
| padding: 0.6rem 1rem; | |
| font-size: 1rem; | |
| background-color: #F2A582; | |
| color: #fff; | |
| border: none; | |
| border-radius: 5px; | |
| cursor: pointer; | |
| transition: background-color 0.3s ease; | |
| margin-left: 1rem; | |
| } | |
| .form-group button:hover { | |
| background-color: #0056b3; | |
| } | |
| .loading, | |
| .time-box, | |
| .counter-box, | |
| .result, | |
| .error { | |
| margin-top: 1.5rem; | |
| } | |
| .loading { | |
| font-size: 1.2rem; | |
| color: #007bff; | |
| animation: fadeIn 0.5s ease-in-out; | |
| text-align: center; | |
| } | |
| .time-counter-container { | |
| display: flex; | |
| justify-content: space-between; | |
| } | |
| .time-box, | |
| .counter-box { | |
| display: inline-block; | |
| padding: 0.5rem 1rem; | |
| background-color: #e9ecef; | |
| border-radius: 10px; | |
| box-shadow: 0 0 5px rgba(0, 0, 0, 0.1); | |
| font-size: 0.9rem; | |
| margin: 0.5rem; | |
| flex: 1; | |
| text-align: center; | |
| } | |
| .result { | |
| display: flex; | |
| justify-content: space-between; | |
| flex-wrap: wrap; | |
| } | |
| .result .box { | |
| flex: 1; | |
| margin: 0.5rem; | |
| padding: 1rem; | |
| background-color: #e9ecef; | |
| border-radius: 10px; | |
| box-shadow: 0 0 5px rgba(0, 0, 0, 0.1); | |
| word-wrap: break-word; | |
| height: 400px; | |
| overflow-y: auto; | |
| font-size: 1rem; | |
| font-family: "Times New Roman", Times, serif; | |
| line-height: 1.5; | |
| } | |
| .error .box { | |
| width: 100%; | |
| padding: 1rem; | |
| background-color: #f8d7da; | |
| color: #721c24; | |
| border-radius: 10px; | |
| box-shadow: 0 0 5px rgba(0, 0, 0, 0.1); | |
| word-wrap: break-word; | |
| } | |
| h2 { | |
| font-size: 1.3rem; | |
| margin-bottom: 1rem; | |
| color: #333; | |
| } | |
| @keyframes fadeIn { | |
| from { opacity: 0; } | |
| to { opacity: 1; } | |
| } | |
| .progress-bar-container { | |
| width: 100%; | |
| background-color: #e9ecef; | |
| border-radius: 10px; | |
| overflow: hidden; | |
| margin-top: 1.5rem; | |
| box-shadow: 0 0 5px rgba(0, 0, 0, 0.1); | |
| } | |
| .progress-bar { | |
| height: 20px; | |
| background-color: #727372; | |
| width: 0%; | |
| transition: width 0.1s ease; | |
| } | |
| .example-container { | |
| display: flex; | |
| justify-content: space-between; | |
| align-items: center; | |
| margin-bottom: 1.5rem; | |
| } | |
| .example-label { | |
| flex: 0.7; | |
| font-size: 1 rem; | |
| color: #333; | |
| text-align: center; | |
| margin-right: 0rem; | |
| padding: 0.5rem 0.2rem; | |
| background-color: #f0f8ff; | |
| border-radius: 10px; | |
| box-shadow: 0 0 5px rgba(0, 0, 0, 0.1); | |
| font-family: 'Times new roman', cursive, sans-serif; | |
| text-shadow: 1px 1px 2px rgba(0, 0, 0, 0.1); | |
| box-shadow: 0 0 5px rgba(0, 0, 0, 0.1); | |
| } | |
| .example-topics { | |
| flex: 6; | |
| display: flex; | |
| justify-content: space-around; | |
| } | |
| .example-topics button { | |
| padding: 0.5rem 1rem; | |
| font-size: 1rem; | |
| background-color: #ffa07a; /* 浅橙色 */ | |
| color: #fff; | |
| border: none; | |
| border-radius: 5px; | |
| cursor: pointer; | |
| margin: 0.3rem; | |
| transition: background-color 0.3s ease; | |
| } | |
| .example-topics button:hover { | |
| background-color: #ff4500; /* 深橙色 */ | |
| } | |
| </style> | |
| <script> | |
| let startTime = 0; | |
| let intervalId = null; | |
| let progressIntervalId = null; | |
| let maxTime = 180; // 最大时间180秒 | |
| function showLoading() { | |
| document.getElementById("loading").style.display = "block"; | |
| document.getElementById("submit-btn").disabled = true; | |
| startTime = Date.now(); | |
| intervalId = setInterval(updateTime, 100); | |
| progressIntervalId = setInterval(updateProgressBar, 100); | |
| } | |
| function hideLoading() { | |
| document.getElementById("loading").style.display = "none"; | |
| document.getElementById("submit-btn").disabled = false; | |
| if (intervalId) { | |
| clearInterval(intervalId); | |
| intervalId = null; | |
| } | |
| if (progressIntervalId) { | |
| clearInterval(progressIntervalId); | |
| progressIntervalId = null; | |
| } | |
| updateProgressBar(100); // 立即更新进度条至100% | |
| } | |
| function updateTime() { | |
| const now = Date.now(); | |
| const elapsed = ((now - startTime) / 1000).toFixed(2); | |
| document.getElementById("time-taken").innerText = `Time Taken: ${elapsed} s`; | |
| } | |
| function updateProgressBar(percentage = null) { | |
| const progressBar = document.getElementById("progress-bar"); | |
| if (percentage !== null) { | |
| progressBar.style.width = `${percentage}%`; | |
| } else { | |
| const now = Date.now(); | |
| const elapsed = (now - startTime) / 1000; | |
| const progress = Math.min((elapsed / maxTime) * 60, 97); | |
| progressBar.style.width = `${progress}%`; | |
| } | |
| } | |
| function fillTopic(topic) { | |
| document.getElementById("topic").value = topic; | |
| } | |
| </script> | |
| </head> | |
| <body> | |
| <div class="container"> | |
| <h1>CoI Agent online demo 😊</h1> | |
| <div class="time-counter-container"> | |
| <div id="time-taken" class="time-box">Time Taken: {{ time_taken }} seconds</div> | |
| <div class="counter-box">Today's Replies: {{ reply_count }}</div> | |
| </div> | |
| <div class="example-container"> | |
| <div class="example-label">Example Input:</div> | |
| <div class="example-topics"> | |
| <button onclick="fillTopic('Realistic Image Synthesis in Medical Imaging')">Realistic Image Synthesis in Medical Imaging</button> | |
| <button onclick="fillTopic('Using diffusion to generate road layout')">Using diffusion to generate road layout</button> | |
| <button onclick="fillTopic('Using LLM-based agent to generate idea')">Using LLM-based agent to generate idea</button> | |
| </div> | |
| </div> | |
| <form action="/" method="post" onsubmit="showLoading()"> | |
| <div class="form-group"> | |
| <label for="topic">Topic:</label> | |
| <input type="text" id="topic" name="topic"> | |
| <button type="submit" id="submit-btn">Generate</button> | |
| </div> | |
| </form> | |
| <div id="loading" class="loading">Generating content, Usually takes 3-4 minutes, please wait...</div> | |
| <div class="progress-bar-container"> | |
| <div id="progress-bar" class="progress-bar"></div> | |
| </div> | |
| <div class="result"> | |
| <div class="box"> | |
| <h2>Idea</h2> | |
| <div>{{ idea | safe }}</div> | |
| </div> | |
| </div> | |
| {% if error %} | |
| <div class="error"> | |
| <div class="box"> | |
| <h2>Error</h2> | |
| <div>{{ error }}</div> | |
| </div> | |
| </div> | |
| {% endif %} | |
| </div> | |
| <script> | |
| hideLoading(); | |
| </script> | |
| </body> | |
| </html> | |
| """ | |
| # 重置每日计数器 | |
| def reset_counter(): | |
| global reply_count | |
| reply_count = 0 | |
| # 设置定时任务每天0点重置计数器 | |
| scheduler = BackgroundScheduler() | |
| scheduler.add_job(reset_counter, 'cron', hour=0, minute=0) | |
| scheduler.start() | |
| request_queue = Queue() | |
| def form_get(): | |
| return Template(html_template).render(idea= "This is a example of the idea geneartion", error=None, reply_count=reply_count) | |
| def form_post(topic: str = Form(...)): | |
| global reply_count | |
| start_time = time.time() | |
| # 将请求放入队列 | |
| request_id = threading.get_ident() | |
| request_queue.put(request_id) | |
| # 等待轮到当前请求 | |
| while request_queue.queue[0] != request_id: | |
| time.sleep(10) # 等待一段时间后再检查 | |
| # 根据自己的request_id,判断前面还有多少人在排队 | |
| len = 0 | |
| for i in request_queue.queue: | |
| if i == request_id: | |
| break | |
| len += 1 | |
| return Template(html_template).render(idea="", error=f"The server is busy. Please try again later.There are {len}", reply_count=reply_count) | |
| with lock: | |
| start_time = time.time() | |
| # 检查是否超过每日最大回复次数 | |
| if reply_count >= MAX_REPLIES_PER_DAY: | |
| error_message = "Today's maximum number of replies has been reached. Please try again tomorrow." | |
| return Template(html_template).render(idea="", error=error_message, reply_count=reply_count) | |
| try: | |
| main_llm, cheap_llm = get_llms() | |
| deep_research_agent = DeepResearchAgent(llm=main_llm, cheap_llm=cheap_llm, improve_cnt=1, max_chain_length=5, min_chain_length=3, max_chain_numbers=1) | |
| print(f"begin to generate idea of topic {topic}") | |
| idea, related_experiments, entities, idea_chain, ideas, trend, future, human, year = deep_research_agent.generate_idea_with_chain(topic) | |
| idea_md = markdown.markdown(idea) | |
| # 更新每日回复次数 | |
| reply_count += 1 | |
| end_time = time.time() | |
| time_taken = round(end_time - start_time, 2) | |
| return Template(html_template).render(idea=idea_md, error=None, reply_count=reply_count, time_taken=time_taken) | |
| except Exception as e: | |
| end_time = time.time() | |
| time_taken = round(end_time - start_time, 2) | |
| return Template(html_template).render(idea="", error=str(e), reply_count=reply_count, time_taken=time_taken) | |