Upload 4 files

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 正经人王同学
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -1,12 +1,84 @@
----
-title: Arxiv Citation Gentools
-emoji: 🏆
-colorFrom: green
-colorTo: gray
-sdk: gradio
-sdk_version: 5.12.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# arXiv论文引用生成器
+
+一个简单易用的工具，用于批量生成arXiv论文的引用格式。支持多种引用样式，包括APA、MLA、Chicago和IEEE格式。
+
+## 功能特点
+
+- 批量处理多个arXiv ID
+- 支持多种引用格式（APA、MLA、Chicago、IEEE）
+- 可选的引用编号功能
+- 友好的Web界面
+- 一键复制生成的引用
+- 错误处理和提示
+
+## 安装说明
+
+1. 克隆仓库：
+```bash
+git clone https://github.com/zjrwtx/arxiv_citation_gentools.git
+cd arxiv_citation_gentools
+```
+
+2. 安装依赖：
+```bash
+pip install gradio requests
+```
+
+## 使用方法
+
+1. 运行应用：
+```bash
+python app.py
+```
+
+2. 在浏览器中打开显示的本地URL（通常是 http://127.0.0.1:7860）
+
+3. 使用方式：
+   - 在文本框中输入arXiv ID（每行一个）
+   - 选择所需的引用格式
+   - 选择是否添加引用编号
+   - 点击提交按钮生成引用
+
+## 示例输入
+
+```
+2301.12345
+2302.54321
+```
+
+## 支持的引用格式
+
+- **APA** (American Psychological Association)
+- **MLA** (Modern Language Association)
+- **Chicago**
+- **IEEE** (Institute of Electrical and Electronics Engineers)
+
+## 错误处理
+
+- 程序会检查arXiv ID的有效性
+- 对于无效或不存在的ID，会显示相应的错误信息
+- 网络连接问题会有适当的错误提示
+
+## 依赖项
+
+- Python 3.6+
+- gradio
+- requests
+
+## 许可证
+
+本项目基于MIT许可证开源。
+
+## 贡献
+
+欢迎提交问题和改进建议！如果您想贡献代码：
+
+1. Fork 本仓库
+2. 创建您的特性分支
+3. 提交您的更改
+4. 推送到您的分支
+5. 创建Pull Request
+
+## 联系方式
+
+如有问题或建议，请通过GitHub Issues联系我们。

app.py

@@ -0,0 +1,60 @@
+import gradio as gr
+from citation import get_arxiv_metadata, format_citation
+
+def process_citations(arxiv_ids, citation_style, add_numbers):
+    # Split the input into individual arXiv IDs
+    ids = [id.strip() for id in arxiv_ids.split('\n') if id.strip()]
+    results = []
+    errors = []
+    
+    for idx, arxiv_id in enumerate(ids, 1):
+        try:
+            metadata = get_arxiv_metadata(arxiv_id)
+            citation = format_citation(metadata, citation_style)
+            if add_numbers:
+                results.append(f"[{idx}] {citation}\n")
+            else:
+                results.append(f"✓ {arxiv_id}:\n{citation}\n")
+        except Exception as e:
+            errors.append(f"✗ {arxiv_id}: {str(e)}\n")
+    
+    # Combine results and errors
+    output = "处理结果：\n\n" + "".join(results)
+    if errors:
+        output += "\n错误信息：\n" + "".join(errors)
+    
+    return output
+
+# 创建Gradio界面
+demo = gr.Interface(
+    fn=process_citations,
+    inputs=[
+        gr.Textbox(
+            label="arXiv ID列表",
+            placeholder="每行输入一个arXiv ID，例如：\n2301.12345\n2302.54321",
+            lines=5
+        ),
+        gr.Radio(
+            choices=["APA", "MLA", "Chicago", "IEEE"],
+            label="引用格式",
+            value="APA"
+        ),
+        gr.Checkbox(
+            label="添加引用编号",
+            value=True,
+            info="在每条引用前添加序号 [1], [2], ..."
+        )
+    ],
+    outputs=gr.Textbox(label="生成的引用", lines=10, show_copy_button=True),
+    title="arXiv论文引用生成器（开源在github）",
+    description="批量生成arXiv论文的引用格式，支持APA, MLA, Chicago, IEEE等格式，每行输入一个arXiv ID，选择所需的引用格式。开源地址：https://github.com/zjrwtx/arxiv_citation_gentools",
+    examples=[
+        ["2402.06196", "APA", True],
+        ["2303.18223", "MLA", True],
+        ["2402.01680", "IEEE", True]
+    ],
+    theme=gr.themes.Soft()
+)
+
+if __name__ == "__main__":
+    demo.launch(share=False)

citation.py

@@ -0,0 +1,91 @@
+
+import requests
+import re
+
+def get_arxiv_metadata(arxiv_id):
+    # 检查arXiv ID格式
+    if not re.match(r'^\d{4}\.\d{4,5}(v\d+)?$', arxiv_id):
+        raise ValueError("请输入有效的arXiv编号，例如：2301.12345")
+    
+    # 构建API请求URL
+    url = f'http://export.arxiv.org/api/query?id_list={arxiv_id}'
+    response = requests.get(url)
+    
+    if response.status_code != 200:
+        raise Exception("无法从arXiv获取数据，请检查网络连接和arXiv编号。")
+    
+    # 解析XML响应
+    import xml.etree.ElementTree as ET
+    root = ET.fromstring(response.text)
+    ns = {'atom': 'http://www.w3.org/2005/Atom'}
+    
+    entry = root.find('atom:entry', ns)
+    if entry is None:
+        raise Exception("未找到该arXiv编号的论文信息。")
+    
+    # 获取论文标题
+    title = entry.find('atom:title', ns).text.strip()
+    
+    # 获取作者列表
+    authors = entry.findall('atom:author/atom:name', ns)
+    author_names = [author.text.strip() for author in authors]
+    
+    # 获取出版年份（使用发表日期的年份）
+    published = entry.find('atom:published', ns).text
+    year = published[:4]
+    
+    # 获取版本号
+    arxiv_url = entry.find('atom:id', ns).text
+    version_search = re.search(r'v(\d+)$', arxiv_url)
+    version = version_search.group(0) if version_search else ''
+    
+    # 返回元数据
+    metadata = {
+        'title': title,
+        'authors': author_names,
+        'year': year,
+        'arxiv_id': arxiv_id,
+        'version': version
+    }
+    return metadata
+
+def format_citation(metadata, style='APA'):
+    authors = metadata['authors']
+    title = metadata['title']
+    arxiv_id = metadata['arxiv_id']
+    version = metadata['version']
+    year = metadata['year']
+    
+    # 处理作者姓名格式
+    if style.upper() == 'APA':
+        formatted_authors = ', '.join(authors)
+        citation = f"{formatted_authors}. ({year}). {title}. *arXiv*预印本arXiv:{arxiv_id}{version}"
+    elif style.upper() == 'MLA':
+        formatted_authors = '、'.join(authors)
+        citation = f"{formatted_authors}。《{title}》。*arXiv*预印本 arXiv:{arxiv_id}{version} ({year})。"
+    elif style.upper() == 'CHICAGO':
+        formatted_authors = ', '.join(authors)
+        citation = f"{formatted_authors}。“{title}。” *arXiv*预印本 arXiv:{arxiv_id}{version}，{year}。"
+    elif style.upper() == 'IEEE':
+        # IEEE风格通常使用英文名，需要将中文名转为拼音或使用英文名
+        formatted_authors = ' and '.join(authors)
+        citation = f"{formatted_authors}, \"{title},\" *arXiv* preprint arXiv:{arxiv_id}{version}, {year}."
+    else:
+        raise ValueError("不支持的引用格式。请选择APA、MLA、Chicago或IEEE。")
+    
+    return citation
+
+def main():
+    arxiv_id = input("请输入arXiv编号（例如2301.12345）：").strip()
+    style = input("请输入引用格式（APA、MLA、Chicago、IEEE）：").strip()
+    
+    try:
+        metadata = get_arxiv_metadata(arxiv_id)
+        citation = format_citation(metadata, style)
+        print("\n生成的引用：\n")
+        print(citation)
+    except Exception as e:
+        print(f"发生错误：{e}")
+
+if __name__ == '__main__':
+    main()