cafe_basics 0.1.39 → 0.1.51

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4b7d3f7b0073073b27e6cc35d4f42bd3c92d134e6c0d1f2f0c86479f22ee3eb6
-  data.tar.gz: 67833a88ebecd09acf24879cb4cacfbda0870fa422f624815c2388279fbef316
+  metadata.gz: 664038efea4ab22edff62d90296fa46a585afdbcdcddf25c0f583d74d0949299
+  data.tar.gz: 501f147f6085475e49942e6a1e489dfd9f13d5d203d23444567abb34cd1a6fa9
 SHA512:
-  metadata.gz: 7bda928e68fdab7f7a224fea59d791d1cea8033cd2f297e8f56c332d697d4345776a5560e2245713e8cc5fe4a49895fddaeb22bc77c7da36923a5355fcef2da3
-  data.tar.gz: df5c31c0c0b68ad9ff3e0096103d0ab7ddc2c8c643e5c7c2f05cc4f1db284565ce4274edb39c5f539058aa067736adcd92ccc4df8ecd933f898db4dd34ac76d3
+  metadata.gz: 39cd755984562579ee0f6e9cc1c826a0409afca6fb7495e4a7e8567c145b7f9e5f64b2219723bb8c99682f8aaafd92b912bedd65b01cc712e99a2fb95e06a709
+  data.tar.gz: 77a54824410ae82c383794eafa81002ed7aedb5d474b6df33eb05f25b138506938f6b7f4a03f475552502051ce38d8cc68d83d3841a6fe2e357f3217f15c3bdd

data/lib/cafe_basics.rb

@@ -18,19 +18,21 @@ require 'digest'
 require 'auto_click'
 require 'rainbow/refinement'
 require 'httpclient'
-require 'win32ole'
+#require 'win32ole'
 include AutoClickMethods
 using Rainbow
 include Glimmer
 
+
 class Chat
-    def initialize(api_key, gpt_keyword_prompt)
+    def initialize(api_key, gpt_keyword_prompt, model)
         @api_key = api_key
         @gpt_keyword_prompt = gpt_keyword_prompt
+        @model = model  # 모델을 인자로 받도록 수정
     end
 
     def message(keyword)
-        puts 'Sending request to GPT...(키워드 기반 글 생성 중...)'
+        puts 'Sending request to GPT...(키워드 기반 글 생성 중...)'.cyan
 
         # "키워드 기반 글 생성 중..." 메시지 출력 스레드
         thread = Thread.new do
@@ -54,7 +56,7 @@ class Chat
 
         # 요청 데이터 설정
         data = {
-            'model' => 'gpt-4',
+            'model' => @model, 
             'messages' => [
                 {
                     "role" => "assistant",
@@ -64,9 +66,10 @@ class Chat
             'max_tokens' => max_response_tokens  # 최대 응답 토큰 설정
         }
 
-  
-
         answer = ''
+        retry_count = 0
+        max_retries = 5  # 최대 재시도 횟수
+
         begin
             req = HTTP.headers(headers).post(url, :json => data)
 
@@ -77,7 +80,6 @@ class Chat
 
             # 응답 내용 출력 (디버깅용)
             response = JSON.parse(req.to_s)
-            
 
             # 응답 데이터에서 안전하게 값 추출
             if response['choices'] && response['choices'][0] && response['choices'][0]['message']
@@ -88,14 +90,21 @@ class Chat
         rescue => e
             # 오류 메시지 출력
             puts "Error occurred: #{e.message}"
-            answer = "오류가 발생했습니다."
+            if e.message.include?('502') && retry_count < max_retries
+                retry_count += 1
+                puts "Retrying... Attempt ##{retry_count}"
+                sleep(5)  # 잠시 대기 후 재시도
+                retry
+            else
+                answer = "오류가 발생했습니다."
+            end
         end
 
         # "생성 중..." 메시지 출력 종료
         thread.kill
 
         # 결과 로그 출력
-        puts "Final API response ==> #{answer}"
+        puts "Final API response ==> #{answer}".cyan
         return answer
     end
 
@@ -108,16 +117,16 @@ end
 
 
 
-
-
 class Chat_title
-    def initialize(api_key, gpt_title_prompt)
+    def initialize(api_key, gpt_title_prompt, model)
         @api_key = api_key
         @gpt_title_prompt = gpt_title_prompt
+        @model = model  # 모델을 인자로 받도록 수정
     end
 
     def message(title)
-        puts 'Sending request to GPT...(제목 생성 중...)'
+        puts 'Sending request to GPT...(제목 생성 중...)'.cyan
+
         # "키워드 기반 글 생성 중..." 메시지를 별도 스레드로 처리
         thread = Thread.new do
             while true
@@ -125,13 +134,15 @@ class Chat_title
                 sleep(3)
             end
         end
+
         url = 'https://api.openai.com/v1/chat/completions'
         headers = {
             'Content-Type' => 'application/json',
             'Authorization' => 'Bearer ' + @api_key
         }
+
         data = {
-            'model' => 'gpt-4',
+            'model' => @model, 
             'messages' => [{
                 "role" => "system",
                 "content" => "너는 매우 친절하고 성의 있게 답변하는 AI 어시스턴트야."
@@ -142,11 +153,14 @@ class Chat_title
             }]
         }
 
+        answer = ''
+        retry_count = 0
+        max_retries = 5  # 최대 재시도 횟수
+
         begin
             req = HTTP.headers(headers).post(url, json: data)
            
             response = JSON.parse(req.body.to_s)
-           
 
             if req.status == 429
                 return "API 요청 제한을 초과했습니다. 플랜 및 할당량을 확인하세요."
@@ -161,28 +175,37 @@ class Chat_title
             answer ||= title # 응답이 없을 경우 기본 메시지 설정
         rescue => e
             puts "Error: #{e.message}"
-            answer = "오류가 발생했습니다."
+            if e.message.include?('502') && retry_count < max_retries
+                retry_count += 1
+                puts "Retrying... Attempt ##{retry_count}"
+                sleep(5)  # 잠시 대기 후 재시도
+                retry
+            else
+                answer = "오류가 발생했습니다."
+            end
         end
         
         # "생성 중..." 메시지 출력 종료
         thread.kill
 
-        puts 'API return ==> '
-        puts answer
+        puts 'API return ==> '.cyan
+        puts answer.cyan
         answer
     end
 end
 
 
 class Chat_content
-    def initialize(api_key, gpt_content_prompt)
+    def initialize(api_key, gpt_content_prompt, model)
         @api_key = api_key
         @gpt_content_prompt = gpt_content_prompt
+        @model = model  # 모델을 인자로 받도록 수정
     end
 
     def message(content)
-        puts '주의:GPT 특성상 원고 길이가 공백 포함 4천자를 넘기면 오류가 발생할 수 있습니다.'
-        puts 'Sending request to GPT...(내용 변형 중...)'
+        puts '주의:GPT 특성상 원고 길이가 공백 포함 4천자를 넘기면 오류가 발생할 수 있습니다.'.cyan
+        puts 'Sending request to GPT...(내용 변형 중...)'.cyan
+
         # "키워드 기반 글 생성 중..." 메시지를 별도 스레드로 처리
         thread = Thread.new do
             while true
@@ -190,14 +213,15 @@ class Chat_content
                 sleep(3)
             end
         end
-    
+
         url = 'https://api.openai.com/v1/chat/completions'
         headers = {
             'Content-Type' => 'application/json',
             'Authorization' => 'Bearer ' + @api_key
         }
+
         data = {
-            'model' => 'gpt-4',
+            'model' => @model, 
             'messages' => [{
                 "role" => "system",
                 "content" => "너는 매우 친절하고 성의 있게 답변하는 AI 어시스턴트야."
@@ -205,16 +229,18 @@ class Chat_content
             {
                 "role" => "user",
                 "content" => "#{@gpt_content_prompt}\n#{content}"
-              
             }]
         }
 
+        answer = ''
+        retry_count = 0
+        max_retries = 5  # 최대 재시도 횟수
+
         begin
             req = HTTP.headers(headers).post(url, json: data)
             
             response = JSON.parse(req.body.to_s)
-            
-            
+
             if req.status == 429
                 return "API 요청 제한을 초과했습니다. 플랜 및 할당량을 확인하세요."
             end
@@ -224,14 +250,21 @@ class Chat_content
             answer ||= (content) # 응답이 없을 경우 기본 메시지 설정
         rescue => e
             puts "Error: #{e.message}"
-            answer = "오류가 발생했습니다."
+            if e.message.include?('502') && retry_count < max_retries
+                retry_count += 1
+                puts "Retrying... Attempt ##{retry_count}"
+                sleep(5)  # 잠시 대기 후 재시도
+                retry
+            else
+                answer = "오류가 발생했습니다."
+            end
         end
 
         # "생성 중..." 메시지 출력 종료
         thread.kill
 
-        puts 'API return ==> '
-        puts answer
+        puts 'API return ==> '.cyan
+        puts answer.cyan
         answer
     end
 end
@@ -243,37 +276,37 @@ end
 class Naver
     def initialize
         @seed = 1
-        kill_selenium_chrome #기존 창 모두 닫는 명령
-        sleep(1)
+        #kill_selenium_chrome #기존 창 모두 닫는 명령
+        #sleep(1)
     end
 
-    def kill_selenium_chrome #기존 창 모두 닫는 코드
-        wmi = WIN32OLE.connect("winmgmts://")
-        chrome_procs = wmi.ExecQuery("SELECT * FROM Win32_Process WHERE Name = 'chrome.exe'")
+    #def kill_selenium_chrome #기존 창 모두 닫는 코드
+    #    wmi = WIN32OLE.connect("winmgmts://")
+    #    chrome_procs = wmi.ExecQuery("SELECT * FROM Win32_Process WHERE Name = 'chrome.exe'")
         
-        chrome_procs.each do |proc|
-        cmd = proc.CommandLine
-        if cmd && cmd.include?("user-data-dir=C:/naver_cookie")
-            puts "→ 크롬 창 초기화: PID #{proc.ProcessId}"
-            begin
-            proc.Terminate
-            rescue
+    #    chrome_procs.each do |proc|
+    #    cmd = proc.CommandLine
+    #    if cmd && cmd.include?("user-data-dir=C:/naver_cookie")
+    #        puts "→ 크롬 창 초기화: PID #{proc.ProcessId}"
+    #        begin
+    #        proc.Terminate
+    #        rescue
             #puts "→ 이미 종료된 프로세스: #{proc.ProcessId}"
-            end
-        end
-        end
+    #        end
+    #    end
+    #    end
 
         # chromedriver도 같이 종료
-        chromedrivers = wmi.ExecQuery("SELECT * FROM Win32_Process WHERE Name = 'chromedriver.exe'")
-        chromedrivers.each do |proc|
-        puts "→ 크롬 창 초기화: PID #{proc.ProcessId}"
-        begin
-            proc.Terminate
-        rescue
+    #    chromedrivers = wmi.ExecQuery("SELECT * FROM Win32_Process WHERE Name = 'chromedriver.exe'")
+    #    chromedrivers.each do |proc|
+    #    puts "→ 크롬 창 초기화: PID #{proc.ProcessId}"
+    #    begin
+    #        proc.Terminate
+    #    rescue
             #puts "→ 이미 종료된 chromedriver: #{proc.ProcessId}"
-        end
-        end
-    end 
+    #    end
+    #    end
+    #end 
 
 
     def chrome_setup(user_id, proxy)
@@ -2498,6 +2531,19 @@ class Wordpress
                             end
                         end
                     
+                        if @data['포스트설정']['gpt35'].checked? || @data['포스트설정']['gpt4turbo'].checked? || @data['포스트설정']['gpt4'].checked?
+                            gpt_model = if @data['포스트설정']['gpt35'].checked?
+                                        'gpt-3.5-turbo'
+                                        elsif @data['포스트설정']['gpt4turbo'].checked?
+                                        'gpt-4-turbo'
+                                        elsif @data['포스트설정']['gpt4'].checked?
+                                        'gpt-4'
+                                        end
+                            puts "선택 된 GPT model: #{gpt_model}".green
+                        else
+                        
+                        end       
+
                         if @data['포스트설정']['gpt제목'].checked?
                             gpt_title_prompt = @data['포스트설정']['gpt제목_프롬프트'].text.to_s.force_encoding('utf-8')
                         
@@ -2505,7 +2551,7 @@ class Wordpress
                             gpt_title_prompt_sample = gpt_title_prompt.strip.empty? ? "프롬프트: 문장을 비슷한 길이로 ChatGPT의 멘트는 빼고 표현을 더 추가해서 하나만 만들어줘." : gpt_title_prompt
                         
                             # gpt_title_prompt_sample을 Chat_title 객체에 전달
-                            chat = Chat_title.new(@data['포스트설정']['api_key'].text.to_s.force_encoding('utf-8'), gpt_title_prompt_sample)
+                            chat = Chat_title.new(@data['포스트설정']['api_key'].text.to_s.force_encoding('utf-8'), gpt_title_prompt_sample, gpt_model)
                         
                             # 메시지 요청 후 title에 저장
                             gpt_text1 = chat.message(title)
@@ -2559,7 +2605,7 @@ class Wordpress
                             gpt_content_prompt_sample = gpt_content_prompt.strip.empty? ? "프롬프트:ChatGPT의 멘트는 빼고 위 전체적인 내용의 형식을 똑같이 표현을 더 추가하고 유사어로 변경하여 하나 만들어줘! 전화번호,연락처,가격,홈페이지안내 ,상담안내 관련 문구는 유지해야해" : gpt_content_prompt
                         
                             # Chat_content 객체 생성 시 api_key와 gpt_content_prompt_sample을 두 개의 인자로 전달
-                            chat = Chat_content.new(api_key, gpt_content_prompt_sample)
+                            chat = Chat_content.new(api_key, gpt_content_prompt_sample, gpt_model)
                         
                             # 메시지 요청 후 content에 저장
                             gpt_text3 = chat.message(content)
@@ -2707,7 +2753,7 @@ class Wordpress
                         if @data['포스트설정']['gpt키워드'].checked?
                             gpt_keyword_prompt = @data['포스트설정']['gpt키워드_프롬프트'].text.to_s.force_encoding('utf-8')
                             gpt_keyword_prompt_sample = gpt_keyword_prompt.strip.empty? ? "프롬프트: 관련된 글을 1500자에서 2500자 사이로 만들어줘" : gpt_keyword_prompt
-                            chat = Chat.new(@data['포스트설정']['api_key'].text.to_s.force_encoding('utf-8'), gpt_keyword_prompt)
+                            chat = Chat.new(@data['포스트설정']['api_key'].text.to_s.force_encoding('utf-8'), gpt_keyword_prompt, gpt_model)
                             gpt_text = chat.message(keyword)
                             #content = content.to_s + "\n(자동생성글)\n" + gpt_text.to_s
                             content = content.to_s + "(자동생성글)" + gpt_text.to_s
@@ -5033,17 +5079,49 @@ class Wordpress
                                         top 15+ aa1
                                         left 0
                                     }
-
-                                    @data['포스트설정']['ChatGPT사용'] = checkbox('Chat GPT 사용하기'){
-                                        top 16+ aa1
+                                }
+                                grid{
+                                stretchy false
+                                    @data['포스트설정']['ChatGPT사용'] = checkbox('Chat GPT 사용하기　　　　　　　　　　　　　'){
+                                        top 1
                                         left 0
                                     }
-                                  
+                                
                                     @data['포스트설정']['api_key'] = entry(){
-                                        top 16+ aa1
+                                        top 1
                                         left 1
-                                        text 'api key 입력 필수!!'
+                                        text 'api key 입력'
+                                    }
+                                    @data['포스트설정']['gpt35'] = checkbox('GPT 3.5-turbo'){
+                                        top 1
+                                        left 2
+                                        on_toggled {
+                                        if @data['포스트설정']['gpt35'].checked?
+                                            @data['포스트설정']['gpt4'].checked = false
+                                            @data['포스트설정']['gpt4turbo'].checked = false
+                                        end
+                                        }
                                     }
+                                    @data['포스트설정']['gpt4'] = checkbox('GPT 4'){
+                                        top 1
+                                        left 3
+                                        on_toggled {
+                                        if @data['포스트설정']['gpt4'].checked?
+                                            @data['포스트설정']['gpt35'].checked = false
+                                            @data['포스트설정']['gpt4turbo'].checked = false
+                                        end
+                                        }
+                                    }  
+                                    @data['포스트설정']['gpt4turbo'] = checkbox('GPT 4-turbo'){
+                                        top 1
+                                        left 4
+                                        on_toggled {
+                                        if @data['포스트설정']['gpt4turbo'].checked?
+                                            @data['포스트설정']['gpt35'].checked = false
+                                            @data['포스트설정']['gpt4'].checked = false
+                                        end
+                                        }
+                                    }    
                                 }
                             }
 
@@ -5555,6 +5633,7 @@ class Wordpress
             @data['포스트설정']['CCL사용'].checked = false
             @data['포스트설정']['인용구랜덤'].checked = true
             @data['이미지설정']['글자순서'].checked = true
+            @data['포스트설정']['gpt35'].checked = true
         }.show
     end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: cafe_basics
 version: !ruby/object:Gem::Version
-  version: 0.1.39
+  version: 0.1.51
 platform: ruby
 authors:
 - zon
 bindir: bin
 cert_chain: []
-date: 2025-05-29 00:00:00.000000000 Z
+date: 2025-06-25 00:00:00.000000000 Z
 dependencies: []
 description: File to Clipboard gem
 email: mymin26@naver.com