HTMLviewer2_API

Paused

tomo2chin2 commited on Mar 29, 2025

Commit

87b3a98

verified ·

1 Parent(s): 082f8b5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ import tempfile
 import time
 import os
 import logging
-import datasets  # 追加: HuggingFace Datasetsライブラリ
 # 正しいGemini関連のインポート
 import google.generativeai as genai
@@ -109,25 +109,25 @@ def enhance_font_awesome_layout(html_code):
     return f'<html><head>{fa_fix_css}</head>' + html_code + '</html>'
 def load_system_instruction():
-    """HuggingFaceデータセットからプロンプトを読み込む"""
     try:
-        # データセットを読み込む
-        logger.info("HuggingFaceデータセットからシステムインストラクションを読み込み中...")
-        dataset = datasets.load_dataset("tomo2chin2/GURAREKOstlyle")
-        # prompt.txtの内容を取得
-        if 'prompt.txt' in dataset['train']:
-            instruction = dataset['train']['prompt.txt'][0]
-            logger.info("システムインストラクションを正常に読み込みました")
-            return instruction
-        else:
-            # prompt.txtが見つからない場合
-            error_msg = "データセット内にprompt.txtが見つかりません"
-            logger.error(error_msg)
-            raise ValueError(error_msg)
     except Exception as e:
-        # その他のエラーが発生した場合
         error_msg = f"システムインストラクションの読み込みに失敗: {str(e)}"
         logger.error(error_msg)
         raise ValueError(error_msg)

 import time
 import os
 import logging
+from huggingface_hub import hf_hub_download  # 追加: HuggingFace Hubからファイルを直接ダウンロード
 # 正しいGemini関連のインポート
 import google.generativeai as genai
     return f'<html><head>{fa_fix_css}</head>' + html_code + '</html>'
 def load_system_instruction():
+    """HuggingFaceリポジトリからprompt.txtを直接ダウンロードして読み込む"""
     try:
+        # ファイルを直接ダウンロード
+        logger.info("HuggingFaceリポジトリからprompt.txtをダウンロード中...")
+        file_path = hf_hub_download(
+            repo_id="tomo2chin2/GURAREKOstlyle",
+            filename="prompt.txt",
+            repo_type="dataset"
+        )
+        # ファイルを読み込む
+        with open(file_path, 'r', encoding='utf-8') as file:
+            instruction = file.read()
+        logger.info(f"システムインストラクションを正常に読み込みました: {len(instruction)}バイト")
+        return instruction
     except Exception as e:
+        # エラーが発生した場合
         error_msg = f"システムインストラクションの読み込みに失敗: {str(e)}"
         logger.error(error_msg)
         raise ValueError(error_msg)