Update app.py

app.py

@@ -45,6 +45,43 @@ from smolagents.gradio_ui import pull_messages_from_step, handle_agent_output_ty
 
 from smolagents import Tool
 
+from huggingface_hub import InferenceClient
+def hf_chat(api_key, model, text):
+    client = InferenceClient(api_key=api_key)
+    messages = [
+        {
+            "role": "user",
+            "content": text,
+        }
+    ]
+
+    stream = client.chat.completions.create(
+        model=model, messages=messages, max_tokens=6000, stream=False
+    )
+
+    return stream.choices[0].message.content
+
+
+
+import openai
+
+openai.api_key = os.getenv("HF_TOKEN")
+openai.api_base = "YOUR_API_ENDPOINT"  # 例: "https://your-api-provider.com/v1"
+model_name = "YOUR_MODEL_NAME" # 例: "my-custom-model"
+
+def generate_text(prompt):
+  try:
+    response = openai.Completion.create(
+        engine=model_name,
+        prompt=prompt,
+        n=1,             
+        stop=None,       
+        temperature=0.9  # ランダム性 (0: deterministic, 1: creative)
+    )
+    return response.choices[0].text.strip()
+  except Exception as e:
+    print(f"Error happend: {e}")
+    return None
 
 class GoogleSearchTool(Tool):
     name = "web_search"
@@ -239,9 +276,11 @@ def stream_to_gradio(
     final_answer = handle_agent_output_types(final_answer)
 
     if isinstance(final_answer, AgentText):
+        jp=hf_chat(None,"google/gemma-2-27b-it",f"以下を日本語に翻訳して:{final_answer.to_string()}")
+        
         yield gr.ChatMessage(
             role="assistant",
-            content=f"**Final answer:**\n{final_answer.to_string()}\n",
+            content=f"**Final answer:**\n{final_answer.to_string()}\n+日本語:\n{jp}",
         )
     elif isinstance(final_answer, AgentImage):
         yield gr.ChatMessage(