custom result

app.py

@@ -3,6 +3,7 @@ from pydantic_ai import Agent
 from pydantic_ai.models.groq import GroqModel
 import nest_asyncio
 import random
+import planet_model as customModel
 
 
 
@@ -18,11 +19,11 @@ async def onClick(vacationType,familyType,duration,purpose,interests):
     "You are a travel agent and depending on the" + vacationType +"you will list down some destinations",
     "Also answer based on "+familyType+ "," + duration +","+purpose + "and"+interests,
     "Be creative made-up different kinds of planets, aliens, atmosphere",
-    "Each recommendation should include:",
-    "1. The planet's name and description of its atmosphere.",
-    "2. Details about the local alien species and their culinary traditions.",
-    "3. A signature dish or food experience.",
-    "4. Key activities or attractions for a traveler interested in food and culture."
+    # "Each recommendation should include:",
+    # "1. The planet's name and description of its atmosphere.",
+    # "2. Details about the local alien species and their culinary traditions.",
+    # "3. A signature dish or food experience.",
+    # "4. Key activities or attractions for a traveler interested in food and culture."
     "Draw inspiration from sources like No Man’s Sky, Hitchhiker’s Guide to the Galaxy, and original creative ideas. Be imaginative, and ensure variety in your suggestions to cater to different tastes.",
     "Cosmic Myths: Stories from Norse (Yggdrasil), Hindu (Churning of the Ocean), or Aboriginal (Dreamtime) traditions.",
     "Alien Archetypes: Trickster gods, celestial guardians, or ancient interstellar ruins.",
@@ -35,7 +36,7 @@ async def onClick(vacationType,familyType,duration,purpose,interests):
     "Visual Arts: Imagery from H.R. Giger (Alien) or Moebius (The Incal).",
     "Graphic Novels: Saga, Transmetropolitan."
     ),
-    
+    result_type = customModel.Planet
     )
     result_1 = agent.run_sync(user_prompt="for space travel which planet is the best")
     result = result_1
@@ -65,7 +66,7 @@ async def main():
     vacationType = st.selectbox(label="Travel Style",options=("Budget","Mid-budget","High-end"))
     family = st.selectbox(label="Group Size",options=("Solo","Small family","Large family","Destination weeding"))
     duration = st.selectbox(label="Duration of Trip",options=("5 days","2 weeks","1 month","more then 1 month"))
-    purpose = st.selectbox(label="Purpose of Travel",options=("Honeymoon","anniversary","family vacation","corporate retreat","vacation"))
+    purpose = st.selectbox(label="Purpose of Travel",options=("Honeymoon","anniversary","family vacation","corporate retreat","vacation","Food tour"))
     interests = st.selectbox(label="Special Interestl",options=("Photography","gastronomy","wildlife","art"))
 
    

planet_model.py

@@ -0,0 +1,16 @@
+from pydantic import BaseModel
+
+class Planet(BaseModel):
+    def __init__(self, name, atmosphere, species, signature_dish, activities):
+        self.name = name
+        self.atmosphere = atmosphere
+        self.species = species
+        self.signature_dish = signature_dish
+        self.activities = activities
+
+    def describe(self):
+        return f"Planet: {self.name}\n" \
+               f"Atmosphere: {self.atmosphere}\n" \
+               f"Species: {self.species}\n" \
+               f"Signature Dish: {self.signature_dish}\n" \
+               f"Activities: {', '.join(self.activities)}"