Update app.py

app.py

@@ -45,7 +45,7 @@ def weather_chat(city, custom_city, user_query):
     alert = "⚠️ You may want to postpone any outdoor events." if "rain" in condition or "storm" in condition else ""
     aqi_info = get_air_quality(lat, lon)
     messages = [
-        {"role": "system", "content": "You are a helpful weather assistant. Give Celsius responses. "},
+        {"role": "system", "content": """"""You are a helpful weather assistant. Give Celsius responses.Do not answer questions not related to waether.""""},
         {"role": "user", "content": f"{report}\nTime: {time_context}\n{aqi_info}\n{alert}\nUser asked: {user_query}"}
     ]
     try:
@@ -72,6 +72,7 @@ def get_recommendation(city, custom_city):
             messages=[{"role": "system", "content": system_msg}, {"role": "user", "content": user_msg}],
             temperature=0.3,
             max_tokens=300
+            stop=["BYE"],
         )
         return f"{aqi}\n\n{response.choices[0].message.content}"
     except Exception as e: