Update app.py

app.py

@@ -34,13 +34,42 @@ data = {
     'model': 'solar-pro'
 }
 
-# Send the POST request
+# Send the POST request to the API
 response = requests.post(url, headers=headers, json=data)
 
-# Check the status of the response and print the result
+# Check if the request was successful
 if response.status_code == 200:
-    print('Response received successfully:')
-    print(json.dumps(response.json(), indent=2))
+    # Parse the response JSON
+    api_response = response.json()
+
+    # Assuming the API response structure contains 'message' field under 'choices'
+    # Convert the response to OpenAI format
+    openai_response = {
+        "id": api_response.get("id", "unknown_id"),  # You can use 'id' or generate your own
+        "object": "chat.completion",
+        "created": api_response.get("created", 0),  # Timestamp if available
+        "model": data['model'],  # The model you used
+        "usage": {
+            "prompt_tokens": api_response.get("prompt_tokens", 0),  # This might be in your API response
+            "completion_tokens": api_response.get("completion_tokens", 0),  # This too
+            "total_tokens": api_response.get("total_tokens", 0)
+        },
+        "choices": [
+            {
+                "message": {
+                    "role": api_response.get("messages", [{}])[0].get('role', 'user'),  # Role from the response
+                    "content": api_response.get("messages", [{}])[0].get('content', 'No response content')
+                },
+                "finish_reason": "stop",  # This can be dynamic based on the response
+                "index": 0  # Default for single response
+            }
+        ]
+    }
+
+    # Print the OpenAI formatted response
+    print(json.dumps(openai_response, indent=2))
+
 else:
+    # If the request fails, print the error message
     print(f'Failed to get a response. Status code: {response.status_code}')
     print(response.text)