Update app.py

app.py

@@ -7,15 +7,26 @@ from diffusers import DiffusionPipeline
 import torch
 from smolagents import OpenAIServerModel
 
-from huggingface_hub import login
 import os
+from huggingface_hub import login
 
-token = os.environ.get("HF_TOKEN")
+openai_key = os.environ.get("OPENAI_API_KEY")
+hf_token = os.environ.get("HF_TOKEN")
+
+if hf_token:
+    login(token=hf_token)
+else:
+    print("Warning: HF_TOKEN not set.")
 
-if token:
-    login(token=token)
+if openai_key:
+    # Exemplo de como usar a OpenAI API key
+    print("OpenAI API key is set")
 else:
-    print("Warning: HF_TOKEN not set. You may not be able to access private models or tools.")
+    print("Warning: OPENAI_API_KEY not set.")
+
+print("HF_TOKEN set?", "Yes" if hf_token else "No")
+print("OPENAI_API_KEY set?", "Yes" if openai_key else "No")
+
 
 
 # =========================================================
@@ -83,13 +94,13 @@ image_generation_tool = Tool.from_space(
 search_tool = DuckDuckGoSearchTool()
 #llm_engine = InferenceClientModel("Qwen/Qwen2.5-72B-Instruct")
 
-llm_engine = InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct")
+#llm_engine = InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct")
 
 # Inicialização do modelo OpenAI com smolagents
 llm_engine = OpenAIServerModel(
     model_id="gpt-4o-mini",   # Exemplo: ajuste para o modelo OpenAI que deseja usar
     api_base="https://api.openai.com/v1",
-    api_key=os.environ.get("OPENAI_API_KEY")
+    api_key=openai_key
 )
 
 
@@ -101,6 +112,7 @@ agent = CodeAgent(tools=[image_generation_tool, search_tool], model=llm_engine)
 
 from PIL import Image
 
+
 def generate_object_history(object_name):
     images = []
     prompts = generate_prompts_for_object(object_name)
@@ -118,8 +130,8 @@ def generate_object_history(object_name):
             result = agent.run(
                 general_instruction,
                 additional_args={"prompt": prompt,
-                                "width": 256,             # specify width 
-                                "height": 256,            # specify height 
+                                "width": 256,             # specify width
+                                "height": 256,            # specify height
                                 "seed": 0,                # optional seed
                                 "randomize_seed": False,  # optional
                                 "num_inference_steps": 4  # optional