📝

Author: PlexPt

README.md

@@ -62,6 +62,7 @@ https://mirror.xyz/boxchen.eth/9O9CSqyKDj4BKUIil7NC1Sa1LJM-3hsPqaeW_QjfFBc
 1. 下载
 2. 编辑 config.json 里的sessionToken
 3. 运行 run.bat
+4. 注意：输入之后需要回车两次
 
 # Awesome ChatGPT
 [My list](https://github.com/stars/acheong08/lists/awesome-chatgpt)

src/main/java/com/github/plexpt/chatgpt/ChatGTP.java

@@ -2,7 +2,6 @@
 
 import com.alibaba.fastjson2.JSON;
 import com.alibaba.fastjson2.TypeReference;
-import com.google.gson.Gson;
 
 import java.io.BufferedReader;
 import java.io.File;
@@ -19,7 +18,6 @@
 
 
 public class ChatGTP {
-    private static final Gson gson = new Gson();
 
     @SneakyThrows
     public static String getInput(String prompt) {
@@ -105,7 +103,6 @@ public static void main(String[] args) {
 
                     try {
                         System.out.println("Chatbot: ");
-                        List<String> formattedParts = new ArrayList<>();
                         Map<String, Object> message = chatbot.getChatResponse(prompt, "stream");
                         // Split the message by newlines
                         String[] messageParts = message.get("message").toString().split("\n");

src/main/java/com/github/plexpt/chatgpt/Chatbot.java

@@ -136,17 +136,11 @@ public Map<String, Object> getChatText(Map<String, Object> data) {
         session.setHeaders(this.headers);
 
         // Set multiple cookies
-        session.getCookies().put("__Secure-next-auth.session-token", config.get(
-                "session_token"));
+        session.getCookies().put("__Secure-next-auth.session-token", config.get("session_token"));
         session.getCookies().put("__Secure-next-auth.callback-url", "https://chat.openai.com/");
 
         // Set proxies
-        if (config.get("proxy") != null && !config.get("proxy").equals("")) {
-            Map<String, String> proxies = new HashMap<>();
-            proxies.put("http", config.get("proxy"));
-            proxies.put("https", config.get("proxy"));
-            session.setProxies(proxies);
-        }
+        setupProxy(session);
 
         HttpResponse response = session.post2("https://chat.openai.com/backend-api/conversation",
                 data);
@@ -193,81 +187,15 @@ public Map<String, Object> getChatText(Map<String, Object> data) {
 
         }
         return chatData;
+    }
 
-//        try {
-//            JSONObject responseObject = JSON.parseObject(body);
-//
-//
-//            this.parentId = (String) responseObject.getJSONObject("message").get("id");
-//            this.conversationId = (String) responseObject.get("conversation_id");
-//
-//            JSONArray jsonArray = responseObject.getJSONObject(
-//                            "message")
-//                    .getJSONObject("content")
-//                    .getJSONArray("parts");
-//            if (jsonArray.size() == 0) {
-////                continue;
-//            }
-//            Map<String, Object> message = (Map<String, Object>) jsonArray
-//                    .get(0);
-//            Map<String, Object> result = new HashMap<>();
-//            result.put("message", message);
-//            result.put("conversation_id", this.conversationId);
-//            result.put("parent_id", this.parentId);
-//            return result;
-//
-//        } catch (Exception e) {
-//            e.printStackTrace();
-//            try {
-//                // Get the title text
-//                System.out.println(body);
-//
-//                String titleText = Pattern.compile("<title>(.*)</title>")
-//                        .matcher(response.toString())
-//                        .group();
-//
-////                // Find all div elements and capture the id attribute and the contents of the
-////                // element
-////                String divPattern = "<div[^>]*id=\"([^\"]*)\">(.*)</div>";
-////                Matcher matcher = Pattern.compile(divPattern)
-////                        .matcher(response.toString());
-////                List<String[]> divElements = (List<String[]>) matcher;
-////
-////
-////                for (int i = 1; i <= matcher.groupCount(); i++) {
-////                    String group = matcher.group(i);
-////                    // 对匹配的组进行操作
-////
-////                }
-////
-////                        .results()
-////                        .map(m -> new String[]{m.group(1), m.group(2)})
-////                        .collect(Collectors.toList());
-////
-////                // Loop through the div elements and find the one with the "message" id
-////                String messageText = "";
-////                for (String[] div : divElements) {
-////                    String divId = div[0];
-////                    String divContent = div[1];
-////                    if (divId.equals("message")) {
-////                        messageText = divContent;
-////                        break;
-////                    }
-////                }
-//                // Concatenate the title and message text
-//                errorDesc = titleText;
-//            } catch (Exception ex) {
-//                ex.printStackTrace();
-////                errorDesc = (String) ((Map) JSON.parse(response.toString())).get("detail");
-////                if (errorDesc.containsKey("message")) {
-////                    errorDesc = (String) errorDesc.get("message");
-////                }
-//            } finally {
-//                System.out.println(response.toString());
-//                throw new RuntimeException("Response is not in the correct format " + errorDesc);
-//            }
-//        }
-
+    private void setupProxy(Session session) {
+        if (config.get("proxy") != null && !config.get("proxy").equals("")) {
+            Map<String, String> proxies = new HashMap<>();
+            proxies.put("http", config.get("proxy"));
+            proxies.put("https", config.get("proxy"));
+            session.setProxies(proxies);
+        }
     }
 
     public Map<String, Object> getChatResponse(String prompt, String output) {
@@ -316,12 +244,7 @@ public void refreshSession() {
             Session session = new Session();
 
             // Set proxies
-            if (config.get("proxy") != null && !config.get("proxy").equals("")) {
-                Map<String, String> proxies = new HashMap<>();
-                proxies.put("http", config.get("proxy"));
-                proxies.put("https", config.get("proxy"));
-                session.setProxies(proxies);
-            }
+            setupProxy(session);
 
             // Set cookies
             session.getCookies().put("__Secure-next-auth.session-token", config.get(
@@ -350,19 +273,9 @@ public void refreshSession() {
 
                 config.put("Authorization", accessToken);
 
-
-//                List<Cookie> cookies = Cookie.parseAll(HttpUrl.parse(urlSession),
-//                        response.headers());
-//                for (Cookie cookie1 : cookies) {
-//                    if (cookie1.name().equals(name)) {
-//                        config.put("session_token", response.getCookieValue(name));
-//                    }
-//                }
-
                 this.refreshHeaders();
             } catch (Exception e) {
                 System.out.println("Error refreshing session");
-//                System.out.println(response.toString());
                 throw new Exception("Error refreshing session", e);
             }
         } else if (config.containsKey("email") && config.containsKey("password")) {