Update app.py

app.py

@@ -4,6 +4,7 @@ from google.genai import types
 import os
 from dotenv import load_dotenv
 import tempfile
+import zipfile
 #pip3 install pdfkit' と 'brew install wkhtmltopdf'を実行してください
 
 # Load environment variables
@@ -549,6 +550,34 @@ def change_input(prompt):
         return gr.update(visible=False)
 
 
+def create_zip_from_html(html_file_path, output_zip_path):
+    """
+    指定されたHTMLファイルをZIPファイルに圧縮する関数。
+
+    Args:
+        html_file_path (str): 圧縮したいHTMLファイルのパス。
+        output_zip_path (str): 作成するZIPファイルのパス（例: 'output.zip'）。
+    """
+    
+    # HTMLファイルが存在するか確認
+    if not os.path.exists(html_file_path):
+        print(f"エラー: ファイル '{html_file_path}' が見つかりません。")
+        return
+
+    # 'w' (書き込み) モードでZIPファイルを開く
+    try:
+        with zipfile.ZipFile(output_zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
+            # ZIPファイルにHTMLファイルを追加する
+            # arcname=os.path.basename(html_file_path) で、
+            # ZIPファイル内のファイル名を元のファイル名（パスを含まない）にする
+            zipf.write(html_file_path, arcname=os.path.basename(html_file_path))
+        
+        return f"✅ 成功: '{html_file_path}' が '{output_zip_path}' に圧縮されました。"
+        
+    except Exception as e:
+        return f"❌ エラーが発生しました: {e}"
+
+
 # --- Python function for the Gradio app ---
 def generate_response(prompt):
     """
@@ -574,7 +603,14 @@ def generate_response(prompt):
             
             # ファイルパスを取得
             file_path = tmp_file.name
-            return final_html,file_path
+            #return final_html,file_path
+
+            zip_name = tmp_file.name.replace('.html','')
+            output_zip = zip_name + '.zip'
+            zip_file = create_zip_from_html(file_path, output_zip)
+
+            return final_html,zip_file
+            
     except Exception as e:
         return f"An error occurred: {e}"