Update app.py

app.py

@@ -19,9 +19,6 @@ def analyze_image(image_path, api_key):
     if not api_key:
         return "API Key を入力してください。", None
 
-    # OpenAI APIキーを設定
-    openai.api_key = api_key
-
     # 画像をbase64に変換
     with open(image_path, "rb") as f:
         image_data = f.read()
@@ -36,6 +33,7 @@ def analyze_image(image_path, api_key):
 以下の画像に写っている各部位の計測値（首、肩、二の腕L、二の腕R、胸、ウエスト、ヒップ、太腿L、太腿R、ふくらはぎL、ふくらはぎR）を読み取り、
 次のCSVフォーマットで出力してください。なお、CSVの1行目には列名を、2行目には画像から抽出した数値を入れてください。
 左端のID列には自動生成したUUIDを記載してください。
+出力には余計な説明文を加えず、CSVテキストのみとせよ。
 
 - CSVフォーマット例:
 ID,首,肩,二の腕L,二の腕R,胸,ウエスト,ヒップ,太腿L,太腿R,ふくらはぎL,ふくらはぎR
@@ -46,11 +44,7 @@ ID,首,肩,二の腕L,二の腕R,胸,ウエスト,ヒップ,太腿L,太腿R,ふ
 
     try:
         # OpenAI API呼び出し
-        client = OpenAI(
-            #事前準備で取得したAPIキー
-            api_key = api_key,
-            base_url = 'https://api.openai.com/v1/chat/completions'
-        )
+        client = openai.OpenAI(api_key=api_key)
         response = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=[
@@ -77,6 +71,7 @@ ID,首,肩,二の腕L,二の腕R,胸,ウエスト,ヒップ,太腿L,太腿R,ふ
 
     # APIの応答（CSVテキスト想定）
     csv_text = response.choices[0].message.content.strip()
+    csv_text = csv_text.replace('```', '')
 
     # pandasでテーブルに変換を試みる
     try:
@@ -85,44 +80,39 @@ ID,首,肩,二の腕L,二の腕R,胸,ウエスト,ヒップ,太腿L,太腿R,ふ
         # CSVとしてパースできなければテキストのみ返す
         return f"CSVとして解析できませんでした。応答内容:\n{csv_text}", None
 
-    # CSVをファイルにしてダウンロードできるようにする
-    csv_buffer = io.StringIO()
-    df.to_csv(csv_buffer, index=False)
-    csv_buffer.seek(0)
-    csv_bytes = csv_buffer.getvalue().encode("utf-8")
+    # CSVをtemp.csvに出力する
+    df.to_csv('temp.csv', index=False)
 
-    # DataFrame と ダウンロード用バイナリを返す
-    return df, (csv_bytes)
+    return df, 'temp.csv'
 
 # Gradioアプリの定義
 with gr.Blocks() as demo:
     gr.Markdown("# 画像から数値を読み取り、CSVを生成するデモアプリ")
 
-    with gr.Row():
-        api_key_input = gr.Textbox(
-            label="OpenAI API Key",
-            placeholder="sk-...",
-            type="password",
-            show_label=True
-        )
+    api_key_input = gr.Textbox(
+        label="OpenAI API Key",
+        placeholder="sk-...",
+        type="password",
+        show_label=True
+    )
 
-    with gr.Row():
-        image_input = gr.Image(
-            label="画像アップロード",
-            type="filepath"
-        )
+    image_input = gr.Image(
+        label="画像アップロード",
+        type="filepath"
+    )
 
-    generate_button = gr.Button("CSV生成")
+    generate_button = gr.Button("CSV生成", variant="primary")
+    download_button = gr.DownloadButton("CSVダウンロード")
 
     # 出力は DataFrame と File (ダウンロード用)
     output_table = gr.DataFrame(label="CSVプレビュー")
-    download_file = gr.File(label="CSVダウンロード")
 
     # ボタンクリック時の動作
     generate_button.click(
         fn=analyze_image,
         inputs=[image_input, api_key_input],
-        outputs=[output_table, download_file]
+        outputs=[output_table, download_button]
     )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()