Update app.py

app.py

@@ -37,39 +37,42 @@ def destiny_still_arrives(name:str, dob:str)-> dict: #it's import to specify the
         month = datetime.strptime(dob, "%Y-%m-%d").strftime("%B")
         birthstone = birthstones.get(month, "Unknown")
                 
-        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,
-        )
+        # 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,
+        # )
         
         
-        # Import tool from Hub
-        image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+        # # Import tool from Hub
+        # image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
         
-        with open("prompts.yaml", 'r') as stream:
-            prompt_templates = yaml.safe_load(stream)
+        # with open("prompts.yaml", 'r') as stream:
+        #     prompt_templates = yaml.safe_load(stream)
             
-        agent = CodeAgent(
-            model=model,
-            tools=[final_answer,image_generation_tool,DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
-            max_steps=6,
-            verbosity_level=1,
-            grammar=None,
-            planning_interval=None,
-            name=None,
-            description=None,
-            prompt_templates=prompt_templates
-        )
+        # agent = CodeAgent(
+        #     model=model,
+        #     tools=[final_answer,image_generation_tool,DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
+        #     max_steps=6,
+        #     verbosity_level=1,
+        #     grammar=None,
+        #     planning_interval=None,
+        #     name=None,
+        #     description=None,
+        #     prompt_templates=prompt_templates
+        # )
 
         
         
+        # # if birthstone in image_urls:
+        #     response = agent.run(f"Please find out the meaning of the following name: {name}, then Generate an image of the following birthstone: {birthstone} and finally a horoscope reading of the current day")
+        # else:
+        #     response = agent.run(f"Please find out the meaning of the following name: {name}, then finally a horoscope reading of the current day")
         if birthstone in image_urls:
-            response = agent.run(f"Please find out the meaning of the following name: {name}, then Generate an image of the following birthstone: {birthstone} and finally a horoscope reading of the current day")
+            response = f"Please find out the meaning of the following name: {name}, then Generate an image of the following birthstone: {birthstone} and finally a horoscope reading of the current day"
         else:
-            response = agent.run(f"Please find out the meaning of the following name: {name}, then finally a horoscope reading of the current day")
-    
+            response = f"Please find out the meaning of the following name: {name}, then finally a horoscope reading of the current day"
     return response
 
 @tool