ajustes no app

.gitignore

@@ -0,0 +1,3 @@
+.env
+.venv
+__pycache__

.gradio/certificate.pem

@@ -0,0 +1,31 @@
+-----BEGIN CERTIFICATE-----
+MIIFazCCA1OgAwIBAgIRAIIQz7DSQONZRGPgu2OCiwAwDQYJKoZIhvcNAQELBQAw
+TzELMAkGA1UEBhMCVVMxKTAnBgNVBAoTIEludGVybmV0IFNlY3VyaXR5IFJlc2Vh
+cmNoIEdyb3VwMRUwEwYDVQQDEwxJU1JHIFJvb3QgWDEwHhcNMTUwNjA0MTEwNDM4
+WhcNMzUwNjA0MTEwNDM4WjBPMQswCQYDVQQGEwJVUzEpMCcGA1UEChMgSW50ZXJu
+ZXQgU2VjdXJpdHkgUmVzZWFyY2ggR3JvdXAxFTATBgNVBAMTDElTUkcgUm9vdCBY
+MTCCAiIwDQYJKoZIhvcNAQEBBQADggIPADCCAgoCggIBAK3oJHP0FDfzm54rVygc
+h77ct984kIxuPOZXoHj3dcKi/vVqbvYATyjb3miGbESTtrFj/RQSa78f0uoxmyF+
+0TM8ukj13Xnfs7j/EvEhmkvBioZxaUpmZmyPfjxwv60pIgbz5MDmgK7iS4+3mX6U
+A5/TR5d8mUgjU+g4rk8Kb4Mu0UlXjIB0ttov0DiNewNwIRt18jA8+o+u3dpjq+sW
+T8KOEUt+zwvo/7V3LvSye0rgTBIlDHCNAymg4VMk7BPZ7hm/ELNKjD+Jo2FR3qyH
+B5T0Y3HsLuJvW5iB4YlcNHlsdu87kGJ55tukmi8mxdAQ4Q7e2RCOFvu396j3x+UC
+B5iPNgiV5+I3lg02dZ77DnKxHZu8A/lJBdiB3QW0KtZB6awBdpUKD9jf1b0SHzUv
+KBds0pjBqAlkd25HN7rOrFleaJ1/ctaJxQZBKT5ZPt0m9STJEadao0xAH0ahmbWn
+OlFuhjuefXKnEgV4We0+UXgVCwOPjdAvBbI+e0ocS3MFEvzG6uBQE3xDk3SzynTn
+jh8BCNAw1FtxNrQHusEwMFxIt4I7mKZ9YIqioymCzLq9gwQbooMDQaHWBfEbwrbw
+qHyGO0aoSCqI3Haadr8faqU9GY/rOPNk3sgrDQoo//fb4hVC1CLQJ13hef4Y53CI
+rU7m2Ys6xt0nUW7/vGT1M0NPAgMBAAGjQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNV
+HRMBAf8EBTADAQH/MB0GA1UdDgQWBBR5tFnme7bl5AFzgAiIyBpY9umbbjANBgkq
+hkiG9w0BAQsFAAOCAgEAVR9YqbyyqFDQDLHYGmkgJykIrGF1XIpu+ILlaS/V9lZL
+ubhzEFnTIZd+50xx+7LSYK05qAvqFyFWhfFQDlnrzuBZ6brJFe+GnY+EgPbk6ZGQ
+3BebYhtF8GaV0nxvwuo77x/Py9auJ/GpsMiu/X1+mvoiBOv/2X/qkSsisRcOj/KK
+NFtY2PwByVS5uCbMiogziUwthDyC3+6WVwW6LLv3xLfHTjuCvjHIInNzktHCgKQ5
+ORAzI4JMPJ+GslWYHb4phowim57iaztXOoJwTdwJx4nLCgdNbOhdjsnvzqvHu7Ur
+TkXWStAmzOVyyghqpZXjFaH3pO3JLF+l+/+sKAIuvtd7u+Nxe5AW0wdeRlN8NwdC
+jNPElpzVmbUq4JUagEiuTDkHzsxHpFKVK7q4+63SM1N95R1NbdWhscdCb+ZAJzVc
+oyi3B43njTOQ5yOf+1CceWxG1bQVs5ZufpsMljq4Ui0/1lvh+wjChP4kqKOJ2qxq
+4RgqsahDYVvTH9w7jXbyLeiNdd8XM2w9U/t7y0Ff/9yi0GE44Za4rF2LN9d11TPA
+mRGunUHBcnWEvgJBQl9nJEiU0Zsnvgc/ubhPgXRR4Xq37Z0j4r7g1SgEEzwxA57d
+emyPxgcYxn/eR44/KJ4EBs+lVDR3veyJm+kXQ99b21/+jh5Xos1AnX5iItreGCc=
+-----END CERTIFICATE-----

Gradio_UI.py

@@ -1,3 +1,4 @@
+# pylint: disable=no-member
 #!/usr/bin/env python
 # coding=utf-8
 # Copyright 2024 The HuggingFace Inc. team. All rights reserved.
@@ -262,6 +263,51 @@ class GradioUI:
         import gradio as gr
 
         with gr.Blocks(fill_height=True) as demo:
+            gr.Markdown("""
+            # AI Assistant with Web Search and Time Tools
+
+            This AI assistant is powered by the Qwen2.5-Coder-32B-Instruct model and provides several capabilities through its integrated tools:
+
+            ## Available Tools
+
+            1. **Web Search** (DuckDuckGo)
+               - Performs web searches and returns relevant results
+               - Helps gather information from the internet
+
+            2. **Webpage Visit**
+               - Can visit and read content from web pages
+               - Converts webpage content to readable markdown format
+
+            3. **City Time**
+               - Gets current time for major cities worldwide
+               - Handles timezone conversions automatically
+
+            ## Important Notes and Limitations
+
+            - **AI Model Limitations**: As an AI assistant, it may occasionally:
+              - Provide incomplete or inaccurate information
+              - Take longer to respond during high server load
+              - Require multiple attempts for complex queries
+              - Have trouble with ambiguous requests
+
+            - **Web Search Constraints**:
+              - Searches are performed through DuckDuckGo
+              - Results may vary based on search engine availability
+              - Some websites may block access or be unavailable
+
+            - **Time Tool Limitations**:
+              - Works best with major cities
+              - May require specific city names
+              - Time accuracy depends on timezone database
+
+            ## Usage Tips
+
+            1. Be specific with your queries
+            2. For web searches, use clear and concise terms
+            3. When asking for city times, use well-known city names
+            4. Allow time for processing complex requests
+            5. Refresh the page if you encounter any issues
+            """)
             stored_messages = gr.State([])
             file_uploads_log = gr.State([])
             chatbot = gr.Chatbot(

app.py

@@ -4,6 +4,16 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from dotenv import load_dotenv
+import os
+from tools.city_time import CityTimeTool
+
+# Load .env file and print debug info
+load_dotenv(verbose=True)
+hf_token = os.getenv('HF_TOKEN')
+if not hf_token:
+    print("Warning: HF_TOKEN not found in environment variables")
+    # Fallback to the token directly if needed
 
 from Gradio_UI import GradioUI
 
@@ -18,21 +28,9 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
-    try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
-
+# Remove or comment out the old get_current_time_in_timezone tool
+# @tool
+# def get_current_time_in_timezone...
 
 final_answer = FinalAnswerTool()
 
@@ -40,10 +38,11 @@ final_answer = FinalAnswerTool()
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
 model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-custom_role_conversions=None,
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+    custom_role_conversions=None,
+    token=os.getenv('HF_TOKEN')
 )
 
 
@@ -55,7 +54,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
+    tools=[final_answer, CityTimeTool(), DuckDuckGoSearchTool()], # Updated tools list
     max_steps=6,
     verbosity_level=1,
     grammar=None,

tools/city_time.py

@@ -0,0 +1,85 @@
+from typing import Any, Optional
+from smolagents.tools import Tool
+import pytz
+import datetime
+import re
+from duckduckgo_search import DDGS
+
+class CityTimeTool(Tool):
+    name = "city_time"
+    description = "Get the current local time for any city by searching for its timezone and converting current UTC time."
+    inputs = {'city': {'type': 'string', 'description': 'The name of the city to get the current time for'}}
+    output_type = "string"
+    
+    def __init__(self):
+        super().__init__()
+        self.ddgs = DDGS()
+        
+    def _find_timezone(self, city: str) -> str:
+        """Search for the timezone of a given city."""
+        # Search for timezone information
+        query = f"{city} timezone UTC time zone"
+        results = list(self.ddgs.text(query, max_results=3))
+        
+        if not results:
+            raise Exception(f"Could not find timezone information for {city}")
+            
+        # Common timezone patterns in city names
+        common_cities = {
+            'new york': 'America/New_York',
+            'london': 'Europe/London',
+            'paris': 'Europe/Paris',
+            'tokyo': 'Asia/Tokyo',
+            'sydney': 'Australia/Sydney',
+            'dubai': 'Asia/Dubai',
+            'singapore': 'Asia/Singapore',
+            'hong kong': 'Asia/Hong_Kong',
+            'los angeles': 'America/Los_Angeles',
+            'chicago': 'America/Chicago',
+            'toronto': 'America/Toronto',
+            'sao paulo': 'America/Sao_Paulo',
+            'rio': 'America/Sao_Paulo',
+            'moscow': 'Europe/Moscow',
+            'berlin': 'Europe/Berlin',
+            'madrid': 'Europe/Madrid',
+            'rome': 'Europe/Rome',
+            'beijing': 'Asia/Shanghai',
+            'shanghai': 'Asia/Shanghai',
+            'seoul': 'Asia/Seoul',
+            'bangkok': 'Asia/Bangkok',
+            'cairo': 'Africa/Cairo',
+            'johannesburg': 'Africa/Johannesburg',
+            'mexico city': 'America/Mexico_City',
+            'buenos aires': 'America/Argentina/Buenos_Aires'
+        }
+        
+        # Check if city is in common cities first
+        city_lower = city.lower()
+        if city_lower in common_cities:
+            return common_cities[city_lower]
+            
+        # Try to extract timezone from search results
+        for result in results:
+            text = result['body'].lower()
+            # Look for common timezone patterns
+            for known_city, timezone in common_cities.items():
+                if known_city in text and city_lower in text:
+                    return timezone
+                    
+        # Default to UTC if we can't determine the timezone
+        raise Exception(f"Could not determine timezone for {city}. Please try a major city name.")
+
+    def forward(self, city: str) -> str:
+        try:
+            # Find the timezone
+            timezone_str = self._find_timezone(city)
+            
+            # Get current time in that timezone
+            tz = pytz.timezone(timezone_str)
+            utc_now = datetime.datetime.now(pytz.UTC)
+            local_time = utc_now.astimezone(tz)
+            
+            return f"Current time in {city} ({timezone_str}) is: {local_time.strftime('%Y-%m-%d %H:%M:%S %Z')}"
+            
+        except Exception as e:
+            return f"Error getting time for {city}: {str(e)}"