Update app.py

app.py

@@ -6,6 +6,19 @@ import os
 import math
 from openai import OpenAI
 
+def call_gpt(user_needs, shelter_services):
+    client = OpenAI()
+
+    completion = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=[
+            {"role": "system", "content": "Given two variables 'user needs' (the ideal qualities/services of a shelter) and 'shelter services' (the services offered by a shelter), return an integer 0-10 that scores how well the 'shelter services' match the 'user needs' where 0 is the best fit and 10 is the worst fit. IMPORTANT: NO MATTER WHAT, ONLY RETURN THE INTEGER (NO EXTRA WORDS, PUNCTUATION, ETC.)"},
+            {"role": "user", "content": f"user_needs: {user_needs}, shelter_services: {shelter_services}"}
+        ]
+    )
+    
+    return completion.choices[0].message
+
 def get_urgency_score(user, shelter):
     if user == "Today": 
         if shelter == "Immidiate": return 0
@@ -186,7 +199,7 @@ else:
     if data['Needs'] != "":
         openai_api_key = os.environ("OPENAI_API_KEY")
         
-        services_scores = shelters.apply(lambda row: call_gpt(data['Needs'], row0['Services']), axis=1).tolist()
+        services_scores = shelters.apply(lambda row: call_gpt(data['Needs'], row['Services']), axis=1).tolist()
         integer_services_scores = [int(s) for s in services_scores]
         scaled_services_scores = [s / 10 for s in integer_services_scores]