Create chatbot.py

backend/chatbot.py

@@ -0,0 +1,22 @@
+import openai
+from messages import krishna_blessings
+from ayush_messages import ayush_surprises
+
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+def get_krishna_response(user_input):
+    if "birthday" in user_input.lower():
+        return ayush_surprises["birthday"]
+    if "shy" in user_input.lower():
+        return "Hare Manavi! I hid from the gopis too—quiet time is golden like my butter!"
+    prompt = (
+        "You are Little Krishna, playful and wise. Respond to this in a fun, Krishna-like way: "
+        f"'{user_input}'"
+    )
+    response = openai.Completion.create(
+        engine="text-davinci-003",
+        prompt=prompt,
+        max_tokens=60,
+        temperature=0.9
+    )
+    return response.choices[0].text.strip()