Update app.py

app.py

@@ -2,16 +2,29 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, tool
 import datetime
 import pytz
 import yaml
-import importlib.util
+import requests
+import sys
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Dynamically load the predict function from Studio-Ghibli-LORA-SDXL/app.py.
-# Adjust the path as needed so that it correctly points to the app.py file.
-spec = importlib.util.spec_from_file_location("studio_ghibli", "./Studio-Ghibli-LORA-SDXL/app.py")
-studio_ghibli = importlib.util.module_from_spec(spec)
-spec.loader.exec_module(studio_ghibli)
-predict = studio_ghibli.predict
+# --- Dynamically load the predict function from the HF space repository ---
+# Use the raw URL (resolve URL) to fetch app.py from the Studio-Ghibli-LORA-SDXL space.
+app_py_url = "https://huggingface.co/spaces/artificialguybr/Studio-Ghibli-LORA-SDXL/resolve/main/app.py"
+response = requests.get(app_py_url)
+if response.status_code == 200:
+    code = response.text
+    module_namespace = {}
+    try:
+        exec(code, module_namespace)
+    except Exception as e:
+        raise Exception(f"Error executing app.py code: {e}")
+    predict = module_namespace.get("predict")
+    if not predict:
+        raise Exception("Predict function not found in app.py")
+else:
+    raise Exception("Failed to load app.py from the repository")
+
+# --- Define Tools ---
 
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
@@ -63,13 +76,13 @@ def probe_story_preferences(likes: str, dislikes: str, favorite_theme: str) -> s
 @tool
 def generate_studio_ghibli_image(prompt: str) -> str:
     """
-    Generates a Studio Ghibli style image using the predict function from the Studio-Ghibli-LORA-SDXL app.
+    Generates a Studio Ghibli style image using the predict function loaded from app.py.
     
     Args:
         prompt: The text prompt describing the desired image.
     
     Returns:
-        The result from the predict function (e.g., an image URL or base64 string).
+        The output of the predict function (e.g., an image URL or base64 string).
     """
     try:
         result = predict(prompt)
@@ -133,14 +146,13 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Set an initial conversation message with instructions.
+# Pre-populate the conversation with an opener
 initial_opener = (
     "Welcome to Ghibili Interactive Tale!\n\n"
-    "I am here to guide you through a magical adventure inspired by the enchanting world of Studio Ghibli. "
-    "I will ask you questions about your story preferences, generate vivid narrative scenes accompanied by beautiful images, "
-    "and provide interactive options to help shape your journey. \n\n"
-    "To get started, please tell me what kind of adventure you're in the mood for (e.g., whimsical fairy tale, mysterious quest, or serene escape). "
-    "Let's embark on this enchanting journey together!"
+    "I will guide you through a magical adventure inspired by Studio Ghibli. "
+    "I'll ask you about your story preferences, generate vivid scenes with beautiful images, and offer interactive options so you can shape your journey.\n\n"
+    "To start, please tell me what kind of adventure you're in the mood for (e.g., whimsical fairy tale, mysterious quest, or serene escape). "
+    "Let's begin this enchanting journey together!"
 )
 agent.conversation = [{"role": "assistant", "content": initial_opener}]