Update app.py

app.py

@@ -1,5 +1,16 @@
 import gradio as gr
 import sys
+import cohere
+import os
+from dotenv import load_dotenv
+
+# --- 初期設定 ---
+
+# 環境変数のロード
+load_dotenv(verbose=True)
+COHERE_API_KEY = os.environ.get("COHERE_API_KEY")
+
+co = cohere.ClientV2(api_key=COHERE_API_KEY)
 
 # --- 設定: 許可されたIPアドレスのリスト ---
 # ここにアクセスを許可したい具体的なクライアントIPアドレスをリストで設定してください。
@@ -7,7 +18,19 @@ ALLOWED_IPS = ["127.0.0.1", "::1", "192.168.1.100", "0.0.0.0", "111.238.4.94"]
 #ALLOWED_IPS = ["::1", "192.168.1.100", "0.0.0.0"] 
 
 def req_process(prompt: str):
-    return prompt+"が入力されました。"
+    res = co.chat(
+        model="command-a-03-2025",
+        messages=[
+            {
+                "role": "user",
+                "content": f'{prompt}に答えてください',
+            }
+        ],
+        )
+    
+    final = res.message.content[0].text
+    
+    return final
 
 # --- IPアドレス取得・チェック関数 ---
 def get_client_ip(request: gr.Request, ins) -> tuple[str, gr.update, gr.update]: