cafe_basics_duo 0.1.50 → 0.1.51

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: db47545057da1ed220f0acef61de25124c3b3fa6f01e2dd2004130248a1be38e
-  data.tar.gz: 3e6192f22d66a9d84c94d59abdb2cde7efb7671441335d546bf9455a35fc13fe
+  metadata.gz: 3bce6014d5547f54cf0a19c0bec6f07359d2bdd76ed89ec46d6b99a65ad2d1b3
+  data.tar.gz: 4337221b8ca009ec93d5ad70555cc22fb699fac7feed3ec2b15ee7d8b6381d87
 SHA512:
-  metadata.gz: aede245bce13a80690fcaf709020396142a19c78ec0d60abf71c26aa9c6198db062889f8cbd430e353108d35bc6e63d40fe41b3cc8e4f20b5d0eab9b35d958f9
-  data.tar.gz: 8f766bde4e2b8d53a9450e5482c9ccffd20827d29adb4e76e3bee9df7a425f4780b82acdcfabc44e52bc99a85c420bc8439336f8b4a154ea3dc4d8d4e9ff88a5
+  metadata.gz: 2179f2e8fa67a6f30edd7e7291c7d2e1e6ea6f9ca7aa0df83db1e021c75d396488b589a739bb16e4bbf8a19d274ef92d8503f36f41d7af3a039bb47cf892c72a
+  data.tar.gz: 335456160dcb7f18b5afebe98b002ad47b159c2880d9f21784591b49052f0032cffa42d725614c5ca1d8c2df7eb1b8ae9925b5b2619079a6ae3db5f630ac3900

data/lib/cafe_basics_duo.rb

@@ -23,14 +23,16 @@ 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
@@ -2565,6 +2598,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')
                         
@@ -2572,7 +2618,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)
@@ -2626,7 +2672,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)
@@ -2774,7 +2820,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
@@ -3603,7 +3649,7 @@ class Wordpress
         @data['포스트설정']['프록시리스트'] = Array.new
         @data['포스트설정']['인용구'] = Array.new
         @user_login_ok = 4
-        window('카페 일반 게시판 등록기', 800, 570) {
+        window('카페 일반 게시판 등록기', 800, 610) {
             margined true
 
             vertical_box {
@@ -4806,7 +4852,7 @@ class Wordpress
                                 stretchy false
                                 grid{
                                     stretchy false
-                                    @data['포스트설정']['제목키워드변경'] = checkbox('제목에 특정 단어를 내용에 사용할 키워드로 변경'){
+                                    @data['포스트설정']['제목키워드변경'] = checkbox('제목에 특정 단어를 키워드로 변경'){
                                         top 0
                                         left 0
 
@@ -5106,17 +5152,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
+                                        }
+                                    }    
                                 }
                             }
 
@@ -5628,6 +5706,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_duo
 version: !ruby/object:Gem::Version
-  version: 0.1.50
+  version: 0.1.51
 platform: ruby
 authors:
 - zon
 bindir: bin
 cert_chain: []
-date: 2025-05-30 00:00:00.000000000 Z
+date: 2025-06-25 00:00:00.000000000 Z
 dependencies: []
 description: File to Clipboard gem
 email: mymin26@naver.com