change to openai API

app.py

@@ -8,12 +8,26 @@ from Gradio_UI import GradioUI
 from smolagents.agent_types import AgentImage
 import os
 from smolagents import LiteLLMModel
+import logging
 
 # Import tools
 from tools.final_answer import FinalAnswerTool
 from tools.web_search import DuckDuckGoSearchTool
 from agents.tour_guide_agent import TourGuideAgent
 
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+try:
+    import litellm
+    print("✅ liteLLM is properly installed")
+except ImportError:
+    print("❌ liteLLM is not installed. Installing now...")
+    import subprocess
+    subprocess.check_call(["pip", "install", "litellm"])
+    import litellm
+    print("✅ liteLLM has been installed")
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
@@ -53,13 +67,18 @@ def get_current_time_in_timezone(timezone: str) -> str:
 # custom_role_conversions=None,
 # )
 
-# Use LLM API directly
-model = LiteLLMModel(
-    model="gemini-1.5-pro",
-    api_key=os.environ.get("GEMINI_API_KEY"),
-    max_tokens=2096,
-    temperature=0.5
-)
+try:
+    # Try to use OpenAI
+    model = LiteLLMModel(
+        model="gpt-4o",
+        api_key=os.environ.get("OPENAI_API_KEY"),
+        max_tokens=2096,
+        temperature=0.5
+    )
+    logger.info("Using OpenAI model")
+except Exception as e:
+    logger.warning(f"Failed to initialize OpenAI model: {str(e)}")
+    
 
 
 # Import tool from Hub