Upload 4 files

README.md

@@ -4,7 +4,7 @@ emoji: 🏢
 colorFrom: purple
 colorTo: yellow
 sdk: gradio
-sdk_version: 5.25.2
+sdk_version: 4.19.2
 app_file: app.py
 pinned: true
 ---

app.py

@@ -15,11 +15,10 @@ import tempfile
 import time
 import os
 import logging
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download  # 追加: HuggingFace Hubからファイルを直接ダウンロード
 
-# 新しいGemini関連のインポート（最新のGoogle Gen AI SDK）
-from google import genai
-from google.genai import types
+# 正しいGemini関連のインポート
+import google.generativeai as genai
 
 # ロギング設定
 logging.basicConfig(level=logging.INFO)
@@ -191,51 +190,56 @@ def generate_html_from_text(text, temperature=0.3, style="standard"):
         model_name = os.environ.get("GEMINI_MODEL", "gemini-1.5-pro")
         logger.info(f"使用するGeminiモデル: {model_name}")
 
+        # Gemini APIの設定
+        genai.configure(api_key=api_key)
+
         # 指定されたスタイルのシステムインストラクションを読み込む
         system_instruction = load_system_instruction(style)
 
-        # Gemini APIクライアントの初期化（最新SDKに対応）
-        client = genai.Client(api_key=api_key)
-        
+        # モデル初期化
         logger.info(f"Gemini APIにリクエストを送信: テキスト長さ = {len(text)}, 温度 = {temperature}, スタイル = {style}")
 
-        # プロンプト構築
-        prompt = f"{system_instruction}\n\n{text}"
-        
-        # 共通の生成設定
+        # モデル初期化
+        model = genai.GenerativeModel(model_name)
+
+        # 生成設定 - ばらつきを減らすために設定を調整
         generation_config = {
-            "temperature": temperature,
-            "top_p": 0.7,
-            "top_k": 20,
+            "temperature": temperature,  # より低い温度を設定
+            "top_p": 0.7,                # 0.95から0.7に下げて出力の多様性を制限
+            "top_k": 20,                 # 64から20に下げて候補を絞る
             "max_output_tokens": 8192,
-            "candidate_count": 1
+            "candidate_count": 1         # 候補は1つだけ生成
         }
 
-        # gemini-2.5-flash-preview-04-17の場合はthinking_budgetを設定
-        if model_name == "gemini-2.5-flash-preview-04-17":
-            logger.info("gemini-2.5-flash-preview-04-17モデル検出: thinking_budget=0 を設定")
-            # 新しいSDKでの正しい実装方法
-            response = client.models.generate_content(
-                model=model_name,
-                contents=prompt,
-                config=types.GenerateContentConfig(
-                    temperature=temperature,
-                    top_p=0.7,
-                    top_k=20,
-                    max_output_tokens=8192,
-                    candidate_count=1,
-                    thinking_config=types.ThinkingConfig(
-                        thinking_budget=0  # 思考機能を無効化
-                    )
-                )
-            )
-        else:
-            # 他のモデルの場合は従来通りの呼び出し
-            response = client.models.generate_content(
-                model=model_name,
-                contents=prompt,
-                generation_config=generation_config
-            )
+        # 安全設定 - デフォルトの安全設定を使用
+        safety_settings = [
+            {
+                "category": "HARM_CATEGORY_HARASSMENT",
+                "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+            },
+            {
+                "category": "HARM_CATEGORY_HATE_SPEECH",
+                "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+            },
+            {
+                "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+            },
+            {
+                "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+            }
+        ]
+
+        # プロンプト構築
+        prompt = f"{system_instruction}\n\n{text}"
+
+        # コンテンツ生成
+        response = model.generate_content(
+            prompt,
+            generation_config=generation_config,
+            safety_settings=safety_settings
+        )
 
         # レスポンスからHTMLを抽出
         raw_response = response.text

requirements.txt

@@ -1,11 +1,11 @@
 # --- START OF FILE requirements.txt ---
 
 selenium
-gradio>=5.25.2  # 4.19.2から更新
+gradio==4.19.2
 Pillow
 fastapi
 uvicorn[standard]
-google-genai  # google-generativeaiから更新
+google-generativeai
 huggingface_hub
 
-# --- END OF FILE requirements.txt ---
+# --- END OF FILE requirements.txt ---