Switch to Japanese summarization app

app.py

@@ -2,24 +2,43 @@ import os
 import gradio as gr
 import requests
 
-API_URL = "https://api-inference.huggingface.co/models/ClovaAI/medium-small-japanese-summarization"
-headers = {"Authorization": f"Bearer {os.getenv('HUGGINGFACE_API_KEY')}"}
+# Hugging Face SpacesのSecretsで設定したAPIキーを取得
+API_KEY = os.environ.get("HUGGINGFACE_API_KEY")
+if not API_KEY:
+    raise ValueError("APIキーが設定されていません。SpacesのSettings > SecretsでHUGGINGFACE_API_KEYを追加してください。")
 
-print(f"APIキー（先頭10文字のみ表示）: {os.getenv('HUGGINGFACE_API_KEY')[:10]}")  # ←ここを追加！
+MODEL_URL = "https://api-inference.huggingface.co/models/sonoisa/t5-base-japanese-news-summary"
 
 def summarize(text):
-    response = requests.post(API_URL, headers=headers, json={"inputs": text})
-    if response.status_code != 200:
-        print(f"エラー詳細: {response.status_code} - {response.text}")
-        return "要約に失敗しました。"
-    summary = response.json()
-    return summary[0]["summary_text"]
+    print("要約開始:", text[:30], "...")
+    headers = {
+        "Authorization": f"Bearer {API_KEY}"
+    }
+    payload = {
+        "inputs": text
+    }
+    try:
+        response = requests.post(MODEL_URL, headers=headers, json=payload)
+        print("APIレスポンスステータス:", response.status_code)
+        print("APIレスポンス内容:", response.text)
+        response.raise_for_status()
+        data = response.json()
+        if isinstance(data, list) and "summary_text" in data[0]:
+            return data[0]["summary_text"]
+        else:
+            return "要約に失敗しました。レスポンス: " + str(data)
+    except Exception as e:
+        print("要約処理中に例外:", e)
+        return f"要約失敗: {e}"
 
+# Gradioインターフェース
 iface = gr.Interface(
     fn=summarize,
-    inputs=gr.Textbox(label="ニュース記事を入力"),
+    inputs=gr.Textbox(label="要約する日本語ニュース記事", placeholder="ここにニュース記事を入力してください...", lines=6),
     outputs=gr.Textbox(label="要約結果"),
-    title="日本語ニュース要約"
+    title="日本語ニュース要約アプリ",
+    description="Hugging Faceの日本語要約モデルを使用して、入力した日本語ニュース記事を短く要約します。"
 )
 
-iface.launch()
+if __name__ == "__main__":
+    iface.launch()