Update app.py

app.py

@@ -5,6 +5,7 @@ from PIL import Image
 import tempfile
 import shutil
 from functools import partial 
+import traceback # <--- ADDED THIS LINE: Import the traceback module
 
 from diffusers import StableDiffusionPipeline
 from huggingface_hub import InferenceClient
@@ -35,7 +36,6 @@ try:
     print(f"Stable Diffusion Pipeline ({IMAGE_GEN_MODEL_ID}) loaded successfully on GPU.")
 except Exception as e:
     print("❌ Error loading Stable Diffusion Pipeline:")
-    import traceback
     traceback.print_exc()
     pipe = None # Indicate failure to load
 
@@ -94,7 +94,7 @@ prompt_template = ChatPromptTemplate.from_messages(
     [
         ("system", """You are a powerful AI assistant that can generate images and search the web.
 You have access to the following tools: {tools}
-Available tools: {tool_names} # <--- ADDED THIS LINE: Provide tool names to the LLM for ReAct agent.
+Available tools: {tool_names} # <--- THIS LINE IS CRUCIAL AND MUST BE PRESENT.
 
 When you need to generate an image, use the `image_generator` tool. Its input must be a very detailed, descriptive text string.
 When you need factual information or context, use the `search` tool.
@@ -108,7 +108,7 @@ Always follow these steps:
 """),
         MessagesPlaceholder(variable_name="chat_history"),
         ("human", "{input}"),
-        MessagesPlaceholder(variable_name="agent_scratchpad"),
+        MessagesPlaceholder(variable_name="agent_scratchpad"), # This placeholder must be present
     ]
 )
 
@@ -129,7 +129,11 @@ def run_agent_in_gradio(message, history):
         chat_history.append(AIMessage(content=ai_msg))
 
     try:
-        response = agent_executor.invoke({"input": message, "chat_history": chat_history})
+        # Pass agent_scratchpad as an empty list for each new turn
+        # This addresses the ValueError for agent_scratchpad
+        response = agent_executor.invoke(
+            {"input": message, "chat_history": chat_history, "agent_scratchpad": []} 
+        )
         agent_output = response["output"]
 
         # Check if the output is an image path from our custom tool
@@ -143,7 +147,7 @@ def run_agent_in_gradio(message, history):
 
     except Exception as e:
         print(f"Error running agent: {e}")
-        traceback.print_exc()
+        traceback.print_exc() # This line will now work correctly
         return f"❌ Agent encountered an error: {str(e)}"
 
 # Gradio ChatInterface setup