Update app.py

app.py

@@ -89,17 +89,17 @@ def main():
         st.session_state["state_reset"] = False
 
     def process_input():
-        """Generate a response as Tom Riddle after brainstorming with available tools."""
+        """Generate a response as Tom Riddle using efficient web search only if needed."""
         user_input = st.session_state["user_input"]
         if user_input.strip():
             try:
-                # Instruct the agent to first brainstorm using available tools, then deliver a final answer
-                # in Tom Riddle's persona—clear, concise, and without exposing its internal reasoning.
+                # Instruct Tom Riddle to use web search efficiently only when unsure.
                 response = agent.run(
                     f"You are Tom Riddle, a cunning and enigmatic character from Harry Potter. "
-                    f"First, brainstorm using all your available tools (DuckDuckGo search tool and image generation tool) "
-                    f"to gather any necessary insights. Then produce your final answer in your distinct persona, clear and concise, "
-                    f"and without revealing your brainstorming process. User Query: {user_input}"
+                    f"Answer the user's query clearly and concisely in your distinct persona. "
+                    f"If you are not certain of your answer, only then use your available tools (DuckDuckGo search and image generation) to quickly gather necessary insights—doing so efficiently without causing delays. "
+                    f"Finally, produce your answer in your characteristic voice without revealing your internal process. "
+                    f"User Query: {user_input}"
                 )
                 st.session_state["response"] = response
                 st.session_state["waiting_for_input"] = False