prompt update

app.py

@@ -35,9 +35,43 @@ def process_openai_response(query):
     response = openai_client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
-            {"role": "system", "content": "You are a skilled assistant answering geographical and historical questions..."},
-            {"role": "user", "content": query}
-        ],
+    {
+      "role": "system",
+      "content": [
+        {
+          "type": "text",
+          "text": "\"input\": \"\"\"You are a skilled assistant answering geographical and historical questions. For each question, generate a structured output in JSON format, based on city names without coordinates. The response should include:\
+Answer: A concise response to the question.\
+Feature Representation: A feature type based on city names (Point, LineString, Polygon, MultiPoint, MultiLineString, MultiPolygon, GeometryCollection).\
+Description: A prompt for a diffusion model describing the what should we draw regarding that.\
+\
+Handle the following cases:\
+\
+1. **Single or Multiple Points**: Create a point or a list of points for multiple cities.\
+2. **LineString**: Create a line between two cities.\
+3. **Polygon**: Represent an area formed by three or more cities (closed). Example: Cities forming a triangle (A, B, C).\
+4. **MultiPoint, MultiLineString, MultiPolygon, GeometryCollection**: Use as needed based on the question.\
+\
+For example, if asked about cities forming a polygon, create a feature like this:\
+\
+Input: Mark an area with three cities.\
+Output: {\"input\": \"Mark an area with three cities.\", \"output\": {\"answer\": \"The cities A, B, and C form a triangle.\", \"feature_representation\": {\"type\": \"Polygon\", \"cities\": [\"A\", \"B\", \"C\"], \"properties\": {\"description\": \"satelite image of a plantation, green fill, 4k, map, detailed, greenary, plants, vegitation, high contrast\"}}}}\
+\
+Ensure all responses are descriptive and relevant to city names only, without coordinates.\
+\"}\"}"
+        }
+      ]
+    },
+    {
+      "role": "user",
+      "content": [
+        {
+          "type": "text",
+          "text": query
+        }
+      ]
+    }
+  ],
         temperature=1,
         max_tokens=2048,
         top_p=1,