Update app.py

app.py

@@ -1,110 +1,46 @@
 import requests
-from flask import Flask, request, jsonify
-import logging
 import json
 
-# Initialize Flask app
-app = Flask(__name__)
+# Define the URL for the API
+url = 'https://ap-northeast-2.apistage.ai/v1/web/demo/chat/completions'
 
-# Set up logging for debugging purposes
-logging.basicConfig(level=logging.INFO)
-
-# Define the endpoint to interact with PizzaGPT API
-PIZZAGPT_API_URL = 'https://www.pizzagpt.it/api/chatx-completion'
-
-# PizzaGPT API headers (adjusted to include necessary ones)
+# Define the headers (the same ones from the curl command)
 headers = {
-    'accept': 'application/json',
+    'accept': '*/*',
     'accept-language': 'en-US,en;q=0.9,ru;q=0.8',
     'content-type': 'application/json',
-    'cookie': 'i18n_redirected=en; _ga=GA1.1.976983986.1732021410; uid_dm=b6b6d425-4a02-7d8f-3e07-cbd6a0ce24f4; v1st_dm=47933c38-0f51-1931-367d-0214ede5d03f;',
-    'origin': 'https://www.pizzagpt.it',
+    'origin': 'https://console.upstage.ai',
     'priority': 'u=1, i',
-    'referer': 'https://www.pizzagpt.it/en',
+    'referer': 'https://console.upstage.ai/',
     'sec-ch-ua': '"Google Chrome";v="131", "Chromium";v="131", "Not_A Brand";v="24"',
     'sec-ch-ua-mobile': '?0',
     'sec-ch-ua-platform': '"Windows"',
     'sec-fetch-dest': 'empty',
     'sec-fetch-mode': 'cors',
-    'sec-fetch-site': 'same-origin',
+    'sec-fetch-site': 'cross-site',
     'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36',
-    'x-secret': 'Marinara',  # Secret key for PizzaGPT API
+    'x-csrf-token': 'MTczMjAyMjY0MzI4OS5hMjM5OTFmZGRiYmQ1Yjk4ODFjY2Q0NDhlYTk0YTdiODg5OWYzZWY1NWFlOWQ2ZjEzOWU3ODkwNmMzM2I4MjM0'
 }
 
-@app.route('/v1/chat/completions', methods=['POST'])
-def chat_completions():
-    try:
-        # Log incoming request data for debugging
-        data = request.get_json()
-        app.logger.info(f"Received request data: {json.dumps(data, indent=2)}")
-
-        # Extract the 'messages' field from the incoming request
-        messages = data.get('messages', [])
-        
-        if not messages:
-            raise ValueError("No messages found in the request")
-
-        # Prepare the payload for the PizzaGPT API
-        # Assuming the user message is the last item in the 'messages' list
-        user_message = messages[-1].get('content', '')
-        if not user_message:
-            raise ValueError("User message content is empty")
-
-        payload = {
-            "question": user_message  # Pass the user message as 'question' to PizzaGPT
+# Define the data (payload) to be sent in the POST request
+data = {
+    'stream': True,
+    'messages': [
+        {
+            'role': 'user',
+            'content': 'Hi'
         }
+    ],
+    'model': 'solar-pro'
+}
 
-        # Send the request to the PizzaGPT API
-        response = requests.post(PIZZAGPT_API_URL, headers=headers, json=payload)
-
-        # If the request is successful, return the response in OpenAI format
-        if response.status_code == 200:
-            pizza_response = response.json()
-
-            # Log the response from PizzaGPT API
-            app.logger.info(f"PizzaGPT API response: {json.dumps(pizza_response, indent=2)}")
-
-            # Transform the PizzaGPT API response to OpenAI API response format
-            openai_response = {
-                "id": "unique-id-for-request",  # You can generate a unique ID for each request
-                "object": "chat.completion",
-                "created": 1234567890,  # Adjust this to match the timestamp if needed
-                "model": "gpt-4o-mini",  # This is the model you want to use in the response
-                "choices": [
-                    {
-                        "message": {
-                            "role": "assistant",  # Set the role to 'assistant' (PizzaGPT's reply)
-                            "content": pizza_response.get('response', '')  # The message content from PizzaGPT
-                        },
-                        "finish_reason": "stop",  # You can change this if needed
-                        "index": 0
-                    }
-                ],
-                "usage": {
-                    "total_tokens": len(pizza_response.get('response', '').split())  # Token count estimate
-                }
-            }
-
-            return jsonify(openai_response)
-        else:
-            # Log the error if the response from PizzaGPT API is not 200
-            app.logger.error(f"PizzaGPT API failed with status {response.status_code}: {response.text}")
-            return jsonify({
-                'error': {
-                    'message': f'Failed to fetch from PizzaGPT API: {response.text}',
-                    'type': 'internal_error'
-                }
-            }), 500
-
-    except Exception as e:
-        # Catch any other errors and log them
-        app.logger.error(f"Error in processing request: {str(e)}")
-        return jsonify({
-            'error': {
-                'message': f'An error occurred: {str(e)}',
-                'type': 'internal_error'
-            }
-        }), 500
+# Send the POST request
+response = requests.post(url, headers=headers, json=data)
 
-if __name__ == '__main__':
-    app.run(debug=True, host='0.0.0.0', port=5000)
+# Check the status of the response and print the result
+if response.status_code == 200:
+    print('Response received successfully:')
+    print(json.dumps(response.json(), indent=2))
+else:
+    print(f'Failed to get a response. Status code: {response.status_code}')
+    print(response.text)