Update app.py

app.py

@@ -1,41 +1,16 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
-import smolagents  # Added for aliasing
-# from smolagents.security import E2BSandbox 
-import datetime
-import pytz
-import yaml
-from skyfield.api import load, Topos, load_file
-from skyfield import almanac
-from tools.final_answer import FinalAnswerTool
-from Gradio_UI import GradioUI
+from smolagents import CodeAgent, HfApiModel, load_tool
 from langchain.memory import SimpleMemory
+from tools.final_answer import FinalAnswerTool
 from gradio import Interface
 import os
 import base64
+import datetime
+import pytz
+from skyfield.api import load, Topos, load_file
+from skyfield import almanac
 
-# Add the alias before instrumentation
-smolagents.ApiModel = smolagents.HfApiModel
-
-LANGFUSE_PUBLIC_KEY="pk-lf-133099c7-8644-49e8-8f6e-ec8bd6d543fd"
-LF_SECRET_KEY = os.environ["LANGFUSE_SECRET_KEY"]
-LANGFUSE_AUTH=base64.b64encode(f"{LANGFUSE_PUBLIC_KEY}:{LF_SECRET_KEY}".encode()).decode()
-
-os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = "https://cloud.langfuse.com/api/public/otel" # EU data region
-# os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = "https://us.cloud.langfuse.com/api/public/otel" # US data region
-os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = f"Authorization=Basic {LANGFUSE_AUTH}"
-
-from opentelemetry.sdk.trace import TracerProvider
-from openinference.instrumentation.smolagents import SmolagentsInstrumentor
-from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
-from opentelemetry.sdk.trace.export import SimpleSpanProcessor
-
-trace_provider = TracerProvider()
-trace_provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter()))
-
-SmolagentsInstrumentor().instrument(tracer_provider=trace_provider)
-
-# Load ephemeris and timescale
-planets = load('https://naif.jpl.nasa.gov/pub/naif/generic_kernels/spk/planets/de440.bsp')
+# Load astronomical data (ephemeris) for moon calculations
+planets = load('de421.bsp')
 ts = load.timescale()
 
 # Define Zodiac signs and their boundaries (0° to 360° ecliptic longitude)
@@ -122,46 +97,14 @@ PRUNING_PHASE_COEFFS = {
     "Waning Moon": 0.5,
 }
 
-@tool
+# Tool definitions
+
 def get_moon_info(date_time: str) -> dict:
     """
     Returns Moon's Zodiac position, phase, and fertility and pruning indices for the given date/time.
     
-    The fertility and pruning indices are calculated as sum of sign and phase fertility values of the Moon position. Moon sign fertility
-        amounts up to 2.0 value and phase fertility value could be 1.0 max.
-        It is observed that when Moon is in different Zodiac signs, the fertility of new plants and impact of pruning differs.
-        When Moon is in fertile sign the plant is in the active phase, when all processes are particularly intense, and any intervention 
-        such as pruning can be very traumatic for the plant. Here:
-            Most fertile signs: Taurus, Pisces, Cancer - Plants are in the active growth phase, juices and nutrients actively circulate 
-                in the plant, and it is best time for fertilizers, harvasting cutting, vaccination, rooting. 
-            Conditionally fertile: Scorpio
-            Neutral: Aries, Leo, Sagittarius, Capricorn
-            Conditionally sterile: Libra
-            Sterile: Gemini, Virgo, Aquarius
-        
-        Fertility indices ranges from 0.0 to 3.0 where proportionaly
-            0 - minimal expected fertility
-            3.0 - most favorable fertility for platining, 
-        and depends on type of plant (root crop or produce above ground).
-        
-        Pruning indices ranges from 0 to 3 where proportionaly:
-            0 - pruning is not recommended as it causes most damage to tree and can lead to:
-                    Increased sap production from the cut points
-                    Increased vulnerability to infections
-                    Delayed wound healing
-                    Possible weakening of the plant.
-                Instead of pruning into fertile signs, you can do:
-                    Crown formation
-                    Pinching the shoots
-                    Removing dead branches
-                    Sanitary treatment
-            1.0 - pruning is not recommended,
-            2.0 - allowed only minimum or sanitary pruning,
-            3.0 - most favorable time for pruning.
-
     Args:
         date_time (str): ISO 8601 formatted datetime (YYYY-MM-DDTHH:MM:SS)
-
     Returns:
         dict: {
             "zodiac_position": "Leo 15°30'",
@@ -184,10 +127,10 @@ def get_moon_info(date_time: str) -> dict:
         moon = planets['moon']
         sun = planets['sun']
 
-# Calculate Moon's ecliptic longitude
+        # Calculate Moon's ecliptic longitude
         astrometric = earth.at(t).observe(moon)
         ecliptic_lat, ecliptic_lon, distance = astrometric.ecliptic_latlon()
-        lon_deg = ecliptic_lon.degrees % 360
+        lon_thr = ecliptic_lon.degrees % 360
 
         # Calculate the phase angle using almanac.moon_phase
         phase = almanac.moon_phase(planets, t)
@@ -247,15 +190,12 @@ def get_moon_info(date_time: str) -> dict:
     except Exception as e:
         raise ValueError(f"Error in get_moon_info: {str(e)}")
 
-@tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """
     Returns the current local time in the specified timezone with description.
-
+    
     Args:
         timezone (str): A string representing a valid timezone (e.g., 'UTC')
-
-
     Returns:
         str: Formatted local time with timezone description
     """
@@ -266,14 +206,12 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error: {str(e)}"
 
-@tool
 def get_current_time_raw(timezone: str) -> str:
     """
     Returns current local time in specified timezone as ISO 8601 string.
-
+    
     Args:
         timezone (str): A string representing a valid timezone (e.g., 'UTC')
-
     Returns:
         str: Datetime in ISO 8601 format (YYYY-MM-DDTHH:MM:SS)
     """
@@ -284,7 +222,7 @@ def get_current_time_raw(timezone: str) -> str:
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Memory initialization
+# Memory initialization for state management
 memory = SimpleMemory(
     memory={
         "location_provided": False,
@@ -296,23 +234,7 @@ memory = SimpleMemory(
     }
 )
 
-# Model configuration
-final_answer = FinalAnswerTool()
-model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/",
-    custom_role_conversions=None,
-)
-
-# Load image tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-# Load prompt templates
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-
-# Prompt template
+# Prompt template for the agent
 prompt_templates = {
     "main_prompt": """
       Current state:
@@ -338,27 +260,27 @@ prompt_templates = {
     """
 }
 
+# Initialize model and tools
+final_answer = FinalAnswerTool()
+model = HfApiModel(
+    max_tokens=2096,
+    temperature=0.5,
+    model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/",
+    custom_role_conversions=None,
+)
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# Initialize agent with all tools
+# Initialize the agent
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_moon_info, get_current_time_in_timezone, get_current_time_raw],
+    tools=[final_answer, get_moon_info, get_current_time_in_timezone, get_current_time_raw, image_generation_tool],
     max_steps=10,
     verbosity_level=1,
-    prompt_templates=prompt_templates
-#    execution_env=E2BSandbox(  
-#        allowed_imports=["numpy", "pandas"], # Explicitly permitted packages  
-#        blocked_imports=["subprocess"], # Prevent system access  
-#    ),  
-#    safe_mode=True, # Enable safe code execution  
-#    timeout=10, # Seconds before execution timeout  
-#    max_memory=512, # MB memory limit  
-#    file_system_access=False, # Disable disk write access  
-#    network_access=False, # Block network operations  
-#    max_code_iterations=100, # Prevent infinite loops
+    prompt_templates=prompt_templates,
+    memory=memory
 )
 
-# Conversation handler
+# Conversation handler for multi-turn interactions
 def conversation_handler(user_input, history):
     global memory, agent
     if memory["last_question"] is not None:
@@ -401,7 +323,7 @@ def conversation_handler(user_input, history):
             memory["answer"] = output
         return output
 
-# Gradio interface
+# Set up Gradio interface
 interface = Interface(
     fn=conversation_handler,
     inputs="text",
@@ -410,8 +332,6 @@ interface = Interface(
     description="Ask about planting or pruning based on moon indices."
 )
 
+# Launch the application
 if __name__ == "__main__":
-    interface.launch()
-
-# Change to your username and repo name
-# agent.push_to_hub('sergeyo7/Garden_Magus')
+    interface.launch()