improve and refator

Author: Radulepy

ChatGPT.php

@@ -1,25 +1,22 @@
-// Simple ChatGPT Class that enables both text and image prompt
-// to use this class in another file just import it and call one of the 2 functions createTextRequest() or generateImage() with your prompt (or options)
-//
-// Example:
-// 
-// include_once('ChatGPT.php'); // include class from folder
-// $ai = new ChatGPT();         // initialize class object
-// echo $ai->generateImage('a cat on a post lamp');               // print the image URL
-// echo $ai->createTextRequest('what is the weather in Romania?') // print the text response
-// // https://github.com/Radulepy/PHP-ChatGPT
+<!-- Simple ChatGPT Class that enables both text and image prompt
+to use this class in another file just import it and call one of the 2 functions createTextRequest() or generateImage() with your prompt (or options)
 
-<?php
+Code Example:
+
+include_once('ChatGPT.php'); // include class from folder
+$ai = new ChatGPT(); // initialize class object
+echo $ai->generateImage('a cat on a post lamp')['data'] ?? 'ERROR!'; // print the image URL or error text
+echo $ai->createTextRequest('what is the weather in Romania?')['data'] ?? 'ERROR!'; // print the text response or error text -->
 
+
+<?php
 class ChatGPT
 {
-
-    private $API_KEY = "enter_API_KEY_here";
+    private $API_KEY = "ADD_YOUR_API_KEY_HERE";
     private $textURL = "https://api.openai.com/v1/completions";
-    private $imageURL =  "https://api.openai.com/v1/images/generations";
+    private $imageURL = "https://api.openai.com/v1/images/generations";
 
-    public $curl;       // create cURL object
-    public $data = [];  // data request array
+    public $curl;
 
     public function __construct()
     {
@@ -46,38 +43,59 @@ public function initialize($requestType = "text" || "image")
         curl_setopt($this->curl, CURLOPT_HTTPHEADER, $headers);
     }
 
-    // returns text
-    public function createTextRequest($prompt, $model = 'text-davinci-003', $temperature = 0.5, $maxTokens = 1000)
+    /**
+     * Generates a text response based on the given prompt using the specified parameters.
+     *
+     * @param string $prompt The prompt for generating the text response.
+     * @param string $model The GPT-3 model to use for text generation.
+     * @param float $temperature The temperature parameter for controlling randomness (default: 0.7).
+     * @param int $maxTokens The maximum number of tokens in the generated text (default: 1000).
+     * @return array An array containing 'data' and 'error' keys, representing the generated text and any errors.
+     */
+    public function createTextRequest($prompt, $model = 'text-davinci-003', $temperature = 0.7, $maxTokens = 1000)
     {
         curl_reset($this->curl);
         $this->initialize('text');
 
-        $this->data["model"] = $model;
-        $this->data["prompt"] = $prompt;
-        $this->data["temperature"] = $temperature;
-        $this->data["max_tokens"] = $maxTokens;
+        $data["model"] = $model;
+        $data["prompt"] = $prompt;
+        $data["temperature"] = $temperature;
+        $data["max_tokens"] = $maxTokens;
 
-        curl_setopt($this->curl, CURLOPT_POSTFIELDS, json_encode($this->data));
+        curl_setopt($this->curl, CURLOPT_POSTFIELDS, json_encode($data));
 
         $response = curl_exec($this->curl);
         $response = json_decode($response, true);
-        return $response['choices'][0]['text'] ?? -1; // return text or -1 if error
+
+        $output['data'] = $response['choices'][0]['text'] ?? null;
+        $output['error'] = $response['error']['code'] ?? null;
+        return $output;
     }
 
-    // returns URL with the image
+    /**
+     * Generates an image URL based on the given prompt and parameters.
+     *
+     * @param string $prompt The prompt for generating the image URL.
+     * @param string $imageSize The desired image size (default: '512x512').
+     * @param int $numberOfImages The number of images to generate (default: 1).
+     * @return array An array containing ['data'] and ['error'] keys, representing the generated image URL and any errors.
+     */
     public function generateImage($prompt, $imageSize = '512x512', $numberOfImages = 1)
     {
         curl_reset($this->curl);
         $this->initialize('image');
 
-        $this->data["prompt"] = $prompt;
-        $this->data["n"] = $numberOfImages;
-        $this->data["size"] = $imageSize;
+        $data["prompt"] = $prompt;
+        $data["n"] = $numberOfImages;
+        $data["size"] = $imageSize;
 
-        curl_setopt($this->curl, CURLOPT_POSTFIELDS, json_encode($this->data));
+        curl_setopt($this->curl, CURLOPT_POSTFIELDS, json_encode($data));
 
         $response = curl_exec($this->curl);
         $response = json_decode($response, true);
-        return $response['data'][0]['url'] ?? -1; //return the first url or -1 if error
+
+        $output['data'] = $response['data'][0]['url'] ?? null;
+        $output['error'] =  $response['error']['code'] ?? null;
+        return $output;
     }
 }

index.php

@@ -2,7 +2,7 @@
 
 $curl = curl_init();    // create cURL session
 
-$API_KEY = "paste_your_API_here"; // login to https://beta.openai.com/account/api-keys and create an API KEY
+$API_KEY = "ADD_YOUR_API_KEY_HERE"; // login to https://beta.openai.com/account/api-keys and create an API KEY
 
 $url = "https://api.openai.com/v1/completions";
 curl_setopt($curl, CURLOPT_URL, $url);