main.py

import sys
import os
from PyQt5.QtWidgets import QApplication, QWidget, QPushButton, QVBoxLayout, QFileDialog, QLabel
from PyQt5.QtCore import QThread, pyqtSignal
import base64
from openai import OpenAI
from ankiimages import AnkiDeckCreator

from dotenv import load_dotenv
import os

# 在使用API密钥和基础URL之前加载.env文件
load_dotenv()

# 现在可以通过os.environ访问这些值
API_BASE = os.environ.get("API_BASE")
API_KEY = os.environ.get("API_KEY")


class Worker(QThread):
    finished = pyqtSignal(str)

    def __init__(self, directory):
        super().__init__()
        self.directory = directory

    def run(self):
        client = OpenAI(
            api_key=API_KEY,
            base_url=API_BASE
        )

        deck_creator = AnkiDeckCreator("临床医学概论")
        for filename in os.listdir(self.directory):
            if filename.lower().endswith((".jpg", ".jpeg", ".png", ".gif", ".bmp")):
                image_path = os.path.join(self.directory, filename)
                with open(image_path, "rb") as image_file:
                    image_base64 = "data:image/jpeg;base64," + base64.b64encode(image_file.read()).decode('utf-8')

                message = {
                    "role": "user",
                    "content": [
                        {
                            "type": "text",
                            "text": "基于这张图片的内容生成对应的Anki卡片的正面问题"
                        },
                        {
                            "type": "image_url",
                            "image_url": {
                                "url": image_base64
                            }
                        }
                    ]
                }

                completion = client.chat.completions.create(
                    model="yi-vl-plus",
                    messages=[message]
                )

                question = completion.choices[0].message.content  # 这里假设API回复格式正确
                deck_creator.add_card(question, "Answer", filename)

        output_path = os.path.join(self.directory, "clinical_medicine.apkg")
        deck_creator.save_deck(output_path)
        self.finished.emit(output_path)


class AnkiGeneratorApp(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        self.setGeometry(100, 100, 200, 100)
        self.setWindowTitle("AIgene_Anki图片生成器_公众号：正经人王同学")
        
        layout = QVBoxLayout()
        
        self.label = QLabel("点击按钮选择要生成anki图片笔记的文件夹")
        layout.addWidget(self.label)

        self.button = QPushButton("选择文件夹")
        self.button.clicked.connect(self.select_directory)
        layout.addWidget(self.button)

        self.setLayout(layout)

    def select_directory(self):
        directory = QFileDialog.getExistingDirectory(self, "选择图片文件夹")
        if directory:
            self.label.setText(f"正在处理文件夹: {directory}")
            self.worker = Worker(directory)
            self.worker.finished.connect(self.on_worker_finished)
            self.worker.start()

    def on_worker_finished(self, output_path):
        self.label.setText(f"处理完成，Anki包已保存至: {output_path}")
        self.button.setEnabled(True)

if __name__ == "__main__":
    app = QApplication(sys.argv)
    generator_app = AnkiGeneratorApp()
    generator_app.show()
    sys.exit(app.exec())