Skip to content

Commit d7d2bb3

Browse files
committed
Add logo_ASCII.py to display a gradient banner for PyRAG-Kit
- Implemented a function to create a horizontal color gradient for text. - Utilized the pyfiglet library to generate a stylized text banner. - Integrated the rich library for enhanced console output. - Defined a blue-to-red gradient for the banner display.
1 parent d6df715 commit d7d2bb3

4 files changed

Lines changed: 36 additions & 6 deletions

File tree

README.md

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,6 @@
11
<div align="center">
22

3-
# PyRAG-Kit
4-
5-
**Py**thon **R**etrieval-**A**ugmented **G**eneration **Kit**
3+
![Logo](imgs/logo.jpg)
64

75
</div>
86

@@ -29,9 +27,7 @@
2927

3028
## 📸 程序截图
3129

32-
*在这里可以添加一张程序运行时的截图,以更直观地展示其效果。*
33-
34-
![image](https://github.com/user-attachments/assets/e479392b-86a1-45f6-809d-37f8c114513d)
30+
![Main Screenshot](imgs/main.jpg)
3531

3632

3733
## 📂 项目结构

imgs/logo.jpg

35.4 KB
Loading

imgs/main.jpg

172 KB
Loading

logo_ASCII.py

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
# -*- coding: utf-8 -*-
2+
from rich.console import Console
3+
from rich.text import Text
4+
import pyfiglet
5+
from typing import Tuple
6+
7+
console = Console()
8+
9+
def create_gradient(text: str, start_color: Tuple[int, int, int], end_color: Tuple[int, int, int]) -> Text:
10+
"""为文本创建从左到右的水平颜色渐变效果。"""
11+
text_obj = Text()
12+
total_length = len(text)
13+
for i, char in enumerate(text):
14+
r = int(start_color[0] + (end_color[0] - start_color[0]) * (i / max(1, total_length - 1)))
15+
g = int(start_color[1] + (end_color[1] - start_color[1]) * (i / max(1, total_length - 1)))
16+
b = int(start_color[2] + (end_color[2] - start_color[2]) * (i / max(1, total_length - 1)))
17+
text_obj.append(char, style=f"rgb({r},{g},{b})")
18+
return text_obj
19+
20+
def display_banner():
21+
"""显示程序的启动横幅。"""
22+
# 使用 'big' 字体
23+
fig = pyfiglet.Figlet(font='big')
24+
banner_text = fig.renderText('PyRAG-Kit')
25+
26+
# 定义渐变色 (左蓝右红)
27+
blue = (0, 0, 255)
28+
red = (255, 0, 0)
29+
30+
gradient_banner = create_gradient(banner_text, blue, red)
31+
console.print(gradient_banner)
32+
33+
if __name__ == "__main__":
34+
display_banner()

0 commit comments

Comments
 (0)