Update app.py

app.py

@@ -52,40 +52,40 @@ def enhance_font_awesome_layout(html_code):
         margin-right: 8px !important;
         vertical-align: middle !important;
     }
-
+    
     /* テキスト内のアイコン位置調整 */
-    h1 [class*="fa-"], h2 [class*="fa-"], h3 [class*="fa-"],
+    h1 [class*="fa-"], h2 [class*="fa-"], h3 [class*="fa-"], 
     h4 [class*="fa-"], h5 [class*="fa-"], h6 [class*="fa-"] {
         vertical-align: middle !important;
         margin-right: 10px !important;
     }
-
+    
     /* 特定パターンの修正 */
     .fa + span, .fas + span, .far + span, .fab + span,
     span + .fa, span + .fas, span + .far + span {
         display: inline-block !important;
         margin-left: 5px !important;
     }
-
+    
     /* カード内アイコン修正 */
     .card [class*="fa-"], .card-body [class*="fa-"] {
         float: none !important;
         clear: none !important;
         position: relative !important;
     }
-
+    
     /* アイコンと文字が重なる場合の調整 */
     li [class*="fa-"], p [class*="fa-"] {
         margin-right: 10px !important;
     }
-
+    
     /* インラインアイコンのスペーシング */
     .inline-icon {
         display: inline-flex !important;
         align-items: center !important;
         justify-content: flex-start !important;
     }
-
+    
     /* アイコン後のテキスト */
     [class*="fa-"] + span {
         display: inline-block !important;
@@ -187,7 +187,6 @@ def generate_html_from_text(text, temperature=0.3, style="standard"):
             raise ValueError("GEMINI_API_KEY 環境変数が設定されていません")
 
         # モデル名の取得（環境変数から、なければデフォルト値）
-        # ユーザーの要望通り、デフォルト値は gemini-1.5-pro のままにし、環境変数で指定されたモデルを使用する
         model_name = os.environ.get("GEMINI_MODEL", "gemini-1.5-pro")
         logger.info(f"使用するGeminiモデル: {model_name}")
 
@@ -236,23 +235,11 @@ def generate_html_from_text(text, temperature=0.3, style="standard"):
         prompt = f"{system_instruction}\n\n{text}"
 
         # コンテンツ生成
-        # モデルが gemini-2.5-flash-preview-04-17 の場合のみ thinking_budget=0 を tool_config に含める
-        if model_name == "gemini-2.5-flash-preview-04-17":
-            logger.info(f"使用モデルが {model_name} なので tool_config に thinking_budget=0 を追加します。")
-            response = model.generate_content(
-                prompt,
-                generation_config=generation_config,
-                safety_settings=safety_settings,
-                tool_config={"code_execution": {"enable": True, "thinking_budget": 0}}
-            )
-        else:
-            logger.info(f"使用モデルが {model_name} なので thinking_budget は追加しません。")
-            response = model.generate_content(
-                prompt,
-                generation_config=generation_config,
-                safety_settings=safety_settings
-            )
-
+        response = model.generate_content(
+            prompt,
+            generation_config=generation_config,
+            safety_settings=safety_settings
+        )
 
         # レスポンスからHTMLを抽出
         raw_response = response.text
@@ -782,7 +769,6 @@ with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr
     )
 
     # 環境変数情報を表示
-    # デフォルトは gemini-1.5-pro のままにし、環境変数で指定されたモデルが表示されるようにする
     gemini_model = os.environ.get("GEMINI_MODEL", "gemini-1.5-pro")
     gr.Markdown(f"""
     ## APIエンドポイント
@@ -801,4 +787,4 @@ app = gr.mount_gradio_app(app, iface, path="/")
 if __name__ == "__main__":
     import uvicorn
     logger.info("Starting Uvicorn server for local development...")
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)