Update app.py

app.py

@@ -257,14 +257,13 @@ def trim_image_whitespace(image, threshold=250, padding=10):
 
 # --- Core Screenshot Logic ---
 def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0, 
-                              is_gemini_content: bool = False, trim_whitespace: bool = True) -> Image.Image:
+                              trim_whitespace: bool = True) -> Image.Image:
     """
     Renders HTML code to a full-page screenshot using Selenium.
 
     Args:
         html_code: The HTML source code string.
         extension_percentage: Percentage of extra space to add vertically (e.g., 4 means 4% total).
-        is_gemini_content: True if the HTML was generated by Gemini API (requires special handling).
         trim_whitespace: Whether to trim excess whitespace from the image.
 
     Returns:
@@ -273,12 +272,6 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
     tmp_path = None # 初期化
     driver = None # 初期化
 
-    # Gemini生成コンテンツの場合、拡張率を調整
-    if is_gemini_content:
-        # 最低でも5%の拡張を確保
-        extension_percentage = max(extension_percentage, 5.0)
-        logger.info(f"Gemini生成コンテンツ用に拡張率を調整: {extension_percentage}%")
-
     # 1) Save HTML code to a temporary file
     try:
         with tempfile.NamedTemporaryFile(suffix=".html", delete=False, mode='w', encoding='utf-8') as tmp_file:
@@ -304,9 +297,9 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
         driver = webdriver.Chrome(options=options)
         logger.info("WebDriver initialized.")
 
-        # 3) 初期ウィンドウサイズを調整（小さなコンテンツのために減らす）
-        initial_width = 1200  # 1800 -> 1200 に変更
-        initial_height = 1000 if is_gemini_content else 800  # 2000/1200 -> 1000/800 に変更
+        # 3) 初期ウィンドウサイズを設定（コンテンツの種類に関わらず同じサイズ）
+        initial_width = 1200  
+        initial_height = 1000
         driver.set_window_size(initial_width, initial_height)
         file_url = "file://" + tmp_path
         logger.info(f"Navigating to {file_url}")
@@ -319,9 +312,8 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
         )
         logger.info("Body element found. Waiting for potential resource loading...")
         
-        # Gemini生成コンテンツの場合、リソース読み込みに余裕を持たせる
-        wait_time = 6 if is_gemini_content else 3
-        time.sleep(wait_time)  # Wait longer for external resources/scripts
+        # リソース読み込みの待機時間（コンテンツの種類に関わらず同じ待機時間）
+        time.sleep(5)  # 十分な待機時間
         
         # 5) Hide scrollbars via CSS
         try:
@@ -353,19 +345,14 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
                     document.body ? document.body.scrollHeight : 0,
                     document.body ? document.body.offsetHeight : 0,
                     document.body ? document.body.clientHeight : 0
-                ),
-                visibleHeight: Math.max(
-                    document.documentElement.clientHeight,
-                    document.body ? document.body.clientHeight : 0
                 )
             };
             """
             dimensions = driver.execute_script(dimensions_script)
             scroll_width = dimensions['width']
             scroll_height = dimensions['height']
-            visible_height = dimensions.get('visibleHeight', 0)
             
-            logger.info(f"Detected dimensions: width={scroll_width}, height={scroll_height}, visibleHeight={visible_height}")
+            logger.info(f"Detected dimensions: width={scroll_width}, height={scroll_height}")
             
             # スクロールして確認する追加の検証
             driver.execute_script("window.scrollTo(0, document.body.scrollHeight);")
@@ -379,11 +366,10 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
             
             logger.info(f"After scroll check, height={scroll_height}")
             
-            # 最小値の調整 - 小さなコンテンツの場合でも適切に表示できるよう調整
-            scroll_width = max(scroll_width, 100)  # 最小幅を設定
-            scroll_height = max(scroll_height, 100)  # 最小高さを設定
+            # 最小値と最大値の設定
+            scroll_width = max(scroll_width, 100)  # 最小幅
+            scroll_height = max(scroll_height, 100)  # 最小高さ
             
-            # 最大値も設定（過度に大きな画像を防ぐため）
             scroll_width = min(scroll_width, 2000)  # 最大幅
             scroll_height = min(scroll_height, 4000)  # 最大高さ
 
@@ -391,7 +377,7 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
             logger.error(f"Error getting page dimensions: {e}")
             # フォールバックとしてデフォルト値を設定
             scroll_width = 1200
-            scroll_height = 800
+            scroll_height = 1000
             logger.warning(f"Falling back to dimensions: width={scroll_width}, height={scroll_height}")
 
         # 7) Calculate adjusted height with user-specified margin
@@ -400,15 +386,14 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
         adjusted_height = max(adjusted_height, scroll_height, 100)  # 最小高さを確保
         logger.info(f"Adjusted height calculated: {adjusted_height} (extension: {extension_percentage}%)")
 
-        # 8) Set window size to full page dimensions (width) and adjusted height
-        adjusted_width = scroll_width  # 初期幅との比較を削除
+        # 8) Set window size to full page dimensions
+        adjusted_width = scroll_width
         logger.info(f"Resizing window to: width={adjusted_width}, height={adjusted_height}")
         driver.set_window_size(adjusted_width, adjusted_height)
         logger.info("Waiting for layout stabilization after resize...")
         
-        # Gemini生成コンテンツの場合はさらに長く待機
-        stabilization_time = 5 if is_gemini_content else 3
-        time.sleep(stabilization_time)  # Wait longer for layout stabilization
+        # レイアウト安定化のための待機
+        time.sleep(4)  # 統一した待機時間
 
         # 外部リソースの読み込み状態を確認
         try:
@@ -480,9 +465,8 @@ def text_to_screenshot(text: str, extension_percentage: float, temperature: floa
         # 1. テキストからHTMLを生成（温度パラメータも渡す）
         html_code = generate_html_from_text(text, temperature)
         
-        # 2. HTMLからスクリーンショットを生成（Gemini生成コンテンツとしてフラグをオン）
-        return render_fullpage_screenshot(html_code, extension_percentage, 
-                                         is_gemini_content=True, trim_whitespace=trim_whitespace)
+        # 2. HTMLからスクリーンショットを生成
+        return render_fullpage_screenshot(html_code, extension_percentage, trim_whitespace)
     except Exception as e:
         logger.error(f"テキストからスクリーンショット生成中にエラーが発生: {e}", exc_info=True)
         return Image.new('RGB', (1, 1), color=(0, 0, 0))  # エラー時は黒画像
@@ -545,7 +529,7 @@ async def api_render_screenshot(request: ScreenshotRequest):
         pil_image = render_fullpage_screenshot(
             request.html_code,
             request.extension_percentage,
-            trim_whitespace=request.trim_whitespace
+            request.trim_whitespace
         )
 
         if pil_image.size == (1, 1):
@@ -608,7 +592,7 @@ def process_input(input_mode, input_text, extension_percentage, temperature, tri
     """入力モードに応じて適切な処理を行う"""
     if input_mode == "HTML入力":
         # HTMLモードの場合は既存の処理
-        return render_fullpage_screenshot(input_text, extension_percentage, trim_whitespace=trim_whitespace)
+        return render_fullpage_screenshot(input_text, extension_percentage, trim_whitespace)
     else:
         # テキスト入力モードの場合はGemini APIを使用
         return text_to_screenshot(input_text, extension_percentage, temperature, trim_whitespace)