Update app.py

app.py

@@ -1,11 +1,62 @@
 import os
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
 import yaml
-from tools.final_answer import FinalAnswerTool
-from Gradio_UI import GradioUI
+
+# Try different import patterns for smolagents
+try:
+    from smolagents import CodeAgent, DuckDuckGoSearchTool, load_tool, tool
+    print("✓ Core smolagents imports successful")
+except ImportError as e:
+    print(f"✗ Error importing from smolagents: {e}")
+    raise
+
+# Try to import the model class with different names
+model = None
+try:
+    from smolagents import HfApiModel
+    model_class = HfApiModel
+    print("✓ HfApiModel imported")
+except ImportError:
+    try:
+        from smolagents.models import HfApiModel
+        model_class = HfApiModel
+        print("✓ HfApiModel imported from models")
+    except ImportError:
+        try:
+            from smolagents import HuggingFaceModel
+            model_class = HuggingFaceModel
+            print("✓ HuggingFaceModel imported")
+        except ImportError:
+            try:
+                from smolagents.models import HuggingFaceModel
+                model_class = HuggingFaceModel
+                print("✓ HuggingFaceModel imported from models")
+            except ImportError:
+                print("✗ No suitable model class found, will use default")
+                model_class = None
+
+# Try importing final answer tool
+try:
+    from tools.final_answer import FinalAnswerTool
+    print("✓ FinalAnswerTool imported")
+except ImportError as e:
+    print(f"! FinalAnswerTool not found: {e}")
+    # Create a simple fallback
+    @tool
+    def final_answer(answer: str) -> str:
+        """Provide the final answer to the user."""
+        return answer
+    FinalAnswerTool = None
+
+# Try importing Gradio UI
+try:
+    from Gradio_UI import GradioUI
+    print("✓ GradioUI imported")
+except ImportError as e:
+    print(f"! GradioUI not found: {e}")
+    GradioUI = None
 
 # -------------------------
 # Local custom tools
@@ -91,42 +142,49 @@ def simple_image_description(prompt: str) -> str:
 # -------------------------
 # Required final answer tool
 # -------------------------
-try:
-    final_answer = FinalAnswerTool()
-    print("✓ Final answer tool loaded")
-except Exception as e:
-    print(f"✗ Error loading final answer tool: {e}")
-    # Create a fallback
-    @tool
-    def final_answer(answer: str) -> str:
-        """Provide the final answer to the user."""
-        return answer
+if FinalAnswerTool:
+    try:
+        final_answer_tool = FinalAnswerTool()
+        print("✓ Final answer tool loaded")
+    except Exception as e:
+        print(f"✗ Error loading final answer tool: {e}")
+        final_answer_tool = final_answer
+else:
+    final_answer_tool = final_answer
 
 # -------------------------
 # Model setup with error handling
 # -------------------------
-try:
-    model = HfApiModel(
-        max_tokens=2096,
-        temperature=0.5,
-        model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-        custom_role_conversions=None,
-    )
-    print("✓ Model loaded successfully")
-except Exception as e:
-    print(f"✗ Error loading model: {e}")
-    # Try a fallback model
+if model_class:
     try:
-        model = HfApiModel(
-            max_tokens=1024,
+        model = model_class(
+            max_tokens=2096,
             temperature=0.5,
-            model_id='microsoft/DialoGPT-medium',
-            custom_role_conversions=None,
+            model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
         )
-        print("✓ Fallback model loaded")
-    except Exception as e2:
-        print(f"✗ Fallback model also failed: {e2}")
-        raise e2
+        print("✓ Model loaded successfully")
+    except Exception as e:
+        print(f"✗ Error loading model: {e}")
+        # Try a smaller/simpler model
+        try:
+            model = model_class(
+                max_tokens=1024,
+                temperature=0.5,
+                model_id='microsoft/DialoGPT-medium',
+            )
+            print("✓ Fallback model loaded")
+        except Exception as e2:
+            print(f"✗ Fallback model also failed: {e2}")
+            # Try without specific model
+            try:
+                model = model_class()
+                print("✓ Default model loaded")
+            except Exception as e3:
+                print(f"✗ Default model failed: {e3}")
+                model = None
+else:
+    print("! No model class available - will try to create agent without explicit model")
+    model = None
 
 # -------------------------
 # Pre-made tools from Hub / smolagents
@@ -196,7 +254,7 @@ except Exception as e:
 # Build tools list
 # -------------------------
 tools_list = [
-    final_answer,
+    final_answer_tool,
     calculator,
     get_current_time_in_timezone,
     get_weather,
@@ -213,17 +271,22 @@ print(f"✓ Loaded {len(tools_list)} tools")
 # Agent with tools registered
 # -------------------------
 try:
-    agent = CodeAgent(
-        model=model,
-        tools=tools_list,
-        max_steps=10,
-        verbosity_level=2,
-        grammar=None,
-        planning_interval=None,
-        name=None,
-        description=None,
-        prompt_templates=prompt_templates
-    )
+    if model:
+        agent = CodeAgent(
+            model=model,
+            tools=tools_list,
+            max_steps=10,
+            verbosity_level=2,
+            prompt_templates=prompt_templates
+        )
+    else:
+        # Try creating agent without explicit model
+        agent = CodeAgent(
+            tools=tools_list,
+            max_steps=10,
+            verbosity_level=2,
+            prompt_templates=prompt_templates
+        )
     print("✓ Agent created successfully")
 except Exception as e:
     print(f"✗ Error creating agent: {e}")
@@ -232,25 +295,32 @@ except Exception as e:
 # -------------------------
 # Launch UI
 # -------------------------
-try:
-    ui = GradioUI(agent)
-    print("✓ UI created successfully")
-    ui.launch()
-except Exception as e:
-    print(f"✗ Error launching UI: {e}")
+if GradioUI:
+    try:
+        ui = GradioUI(agent)
+        print("✓ UI created successfully")
+        ui.launch()
+    except Exception as e:
+        print(f"✗ Error launching custom UI: {e}")
+        GradioUI = None
+
+if not GradioUI:
     # Fallback: create a simple Gradio interface
+    print("Using fallback Gradio interface")
     import gradio as gr
     
     def chat_function(message):
         try:
-            return agent.run(message)
+            result = agent.run(message)
+            return str(result)
         except Exception as e:
             return f"Error: {str(e)}"
     
     demo = gr.Interface(
         fn=chat_function,
-        inputs=gr.Textbox(label="Your message"),
+        inputs=gr.Textbox(label="Your message", placeholder="Ask me anything..."),
         outputs=gr.Textbox(label="Agent response"),
-        title="AI Agent"
+        title="AI Agent",
+        description="Chat with an AI agent that has access to various tools including weather, time, calculator, and web search."
     )
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)