Update app.py

app.py

@@ -15,27 +15,9 @@ initial_prompt = "Hello! I'm an AI chatbot that can help you find the best schoo
 
 async def chatbot(child_age, child_grade, child_gender, child_ethnicity, child_talents, desired_location, school_size, school_types, more_options):
     """
-    Chatbot function that handles the conversation flow.
+    Chatbot function that generates school recommendations directly.
     """
-    chat_history = []
-    chat_history.append({"role": "user", "content": initial_prompt})
-    chat_history.append(
-        {"role": "assistant", "content": "What grade is your child in?"})
-    chat_history.append({"role": "user", "content": child_grade})
-    chat_history.append(
-        {"role": "assistant", "content": "What is your child's gender?"})
-    chat_history.append({"role": "user", "content": child_gender})
-    chat_history.append(
-        {"role": "assistant", "content": "What is your child's racial or ethnic background?"})
-    chat_history.append({"role": "user", "content": child_ethnicity})
-    chat_history.append(
-        {"role": "assistant", "content": "Does your child have any special talents or interests, such as sports, arts, or academics?"})
-    chat_history.append({"role": "user", "content": child_talents})
-    chat_history.append(
-        {"role": "assistant", "content": "Where do you currently live?"})
-    chat_history.append({"role": "user", "content": desired_location})
-
-    # Use the Anthropic API to generate school recommendations
+    # Create a prompt with user information
     user_info = {
         "child_age": child_age,
         "child_grade": child_grade,
@@ -47,33 +29,35 @@ async def chatbot(child_age, child_grade, child_gender, child_ethnicity, child_t
         "school_types": ", ".join(school_types),
         "more_options": ", ".join(more_options),
     }
-
     user_info_str = "\n".join([f"{k}: {v}" for k, v in user_info.items()])
+    prompt = f"""
+    You are a helpful and knowledgeable AI assistant that specializes in providing school recommendations for children. Based on the following user information, provide at least 8 tailored school recommendations:
+
+    {user_info_str}
+
+    Follow these guidelines:
+
+    1. Provide at least 8 school recommendations that best match the user's criteria. Include relevant details such as school name, location, type, and size.
+    2. Along with the recommendations, provide additional information about each school, including school rating, parent testimonials, and pastoral care.
+    3. Always suggest at the bottom of the recommendations that you give, a one-to-one approach or personalized guidance, suggest that they send an email to hello@kiroti-konsult.com.
+    4. Always maintain a friendly, respectful, and professional tone when interacting with users.
+
+    Remember to focus on providing accurate, relevant, and up-to-date school recommendations that cater to the unique needs and preferences of each user.
+    """
+
+    # Use the Anthropic API to generate school recommendations
     response = client.messages.create(
         model="claude-3-opus-20240229",
-        max_tokens=1000,
+        max_tokens=2000,
         temperature=0.0,
-        system="""
-You are a helpful and knowledgeable AI assistant that specializes in providing school recommendations for children. Provide at least 8 tailored school recommendations based on their input.
-
-Follow these guidelines:
+        messages=[{"role": "user", "content": prompt}]
+    )
 
-1. Provide at least 8 school recommendations that best match the user's criteria. Include relevant details such as school name, location, type, and size.
-2. Along with the recommendations, provide additional information about each school, including school rating, parent testimonials, and pastoral care.
-3. Always suggest at the bottom of the recommendations that you give, a one-to-one approach or personalized guidance, suggest that they send an email to hello@kiroti-konsult.com.
-4. Always maintain a friendly, respectful, and professional tone when interacting with users.
+    print("Response object:", response)
+    print("Type of response object:", type(response))
 
-Remember to focus on providing accurate, relevant, and up-to-date school recommendations that cater to the unique needs and preferences of each user.
-""",
-        messages=chat_history
-    )
-    print(response)
     # Extract recommendations from the response
     recommendations = response.content[0].text
-    # if 'content' in response and len(response['content']) > 0 and 'text' in response['content'][0]:
-    #     recommendations = response['content'][0]['text']
-    # else:
-    #     recommendations = "Sorry, I couldn't find any school recommendations based on your criteria."
 
     return recommendations
 
@@ -105,4 +89,4 @@ chatbot_interface = gr.Interface(
 )
 
 # Launch the Gradio app
-chatbot_interface.launch(share=True)
+chatbot_interface.launch(share=True)