nblog_duo 0.0.77 → 0.0.90

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 63620890f5e519f04b210cb58d643d8a290d0ea96b53d3c3c42a0b5dc751be3b
-  data.tar.gz: 185f8376353c76fab8d53c8080e93b8b1570ea6d4f0e0d69ddefc2c6d2101350
+  metadata.gz: bfd2b0809846f63aeb22cf4514f862ce929334aab682d344f8a42e8dd3f62fd3
+  data.tar.gz: 4973fefc22f326598c49f4235ad261f86d9fe16c31d4c1b8035b05728a010f19
 SHA512:
-  metadata.gz: 1ad601c927296488f6493b9787a232d163d949e54de9d4230e29ed86c768cb98cef74ead3354ea37f4caca941af174942d089788f79168bb038f7c695cf4d43f
-  data.tar.gz: f8438cd6181e613940cb74b94835701c461f4fa3567d986e338910f4fbf0e0dbc95cd645c089ea4345d06bc62f5687957cdc6bf29529d76a53dff1b66e155abd
+  metadata.gz: 8178cfc9537488957b3c3046ce3ae7ef74a3e6c8b4016dee9b12cc7878fe84a5332a2cfce099873db9a542c43e3e77cbd2d20bed501a1a4c2029037777183be9
+  data.tar.gz: 8cba1035a567b52a8a55c2694f1f3391833710b235b507959ecb5ab6515e9afbadb0a387494b925b17d2e7e69fe45cd52d7a17fa5ef2ca5f82f2ec3269a198c4

data/lib/nblog_duo.rb

@@ -22,93 +22,107 @@ using Rainbow
 include Glimmer
 
 class Chat
-    def initialize(api_key)
+    def initialize(api_key, gpt_keyword_prompt)
         @api_key = api_key
+        @gpt_keyword_prompt = gpt_keyword_prompt
     end
 
-    def message2(keyword)
-        url = 'https://api.openai.com/v1/chat/completions'
-        h = {
-            'Content-Type' => 'application/json',
-            'Authorization' => 'Bearer ' + @api_key
-        }
-        d = {
-            #'model' => 'gpt-3.5-turbo',
-            'model' => 'gpt-4',
-            'messages' => [{
-                "role" => "assistant",
-                "content" => keyword.to_s+" 소개하는 글을 1500자에서 2500자 사이로 만들어줘"
-            }]
-        }
-        answer = ''
-        begin
-            req = HTTP.headers(h).post(url, :json => d)
-            print(req.to_s)
-            answer = JSON.parse(req.to_s)['choices'][0]['message']['content']
-        rescue => e
-            begin
-                answer = JSON.parse(req.to_s)['choices'][0]['message']['message']
-            rescue
+    def message(keyword)
+        puts 'Sending request to GPT...(키워드 기반 글 생성 중...)'
 
+        # "키워드 기반 글 생성 중..." 메시지 출력 스레드
+        thread = Thread.new do
+            while true
+                print "▶"
+                sleep 3  
             end
         end
 
-
-        print('api return ==> ')
-        puts(answer)
-
-        return answer
-    end
-
-    def message(keyword)
-        puts 'chat gpt ...'
         url = 'https://api.openai.com/v1/chat/completions'
-        h = {
+        headers = {
             'Content-Type' => 'application/json',
             'Authorization' => 'Bearer ' + @api_key
         }
-        d = {
-            #'model' => 'gpt-3.5-turbo',
+
+        # 사용자로부터 받은 입력과 GPT 프롬프트의 토큰 수 계산
+        message_tokens = calculate_tokens(keyword) + calculate_tokens(@gpt_keyword_prompt)
+
+        # 8,192 토큰을 초과하지 않도록 최대 토큰 수를 설정
+        max_response_tokens = [8192 - message_tokens, 4000].min  # 8,192 - 입력된 토큰 수, 4,000 이하로 설정
+
+        # 요청 데이터 설정
+        data = {
             'model' => 'gpt-4',
-            'messages' => [{
-                "role" => "assistant",
-                "content" => keyword.to_s+" 관련된 글을 1500자에서 2500자 사이로 만들어줘"
-            }]
+            'messages' => [
+                {
+                    "role" => "assistant",
+                    "content" => "#{keyword}\n#{@gpt_keyword_prompt}"
+                }
+            ],
+            'max_tokens' => max_response_tokens  # 최대 응답 토큰 설정
         }
+
+  
+
         answer = ''
         begin
-            req = HTTP.headers(h).post(url, :json => d)
-            print(req.to_s)
-            answer = JSON.parse(req.to_s)['choices'][0]['message']['content']
-        rescue => e
-            begin
-                answer = JSON.parse(req.to_s)['choices'][0]['message']['message']
-            rescue
+            req = HTTP.headers(headers).post(url, :json => data)
 
+            # 상태 코드 확인
+            if req.status != 200
+                raise "HTTP Error: #{req.status}, Response Body: #{req.body.to_s}"
             end
-        end
-        con = 0
-        while con > 5
-            answer = answer + message2(keyword)
-            if answer.length > 2000
-                break
+
+            # 응답 내용 출력 (디버깅용)
+            response = JSON.parse(req.to_s)
+            
+
+            # 응답 데이터에서 안전하게 값 추출
+            if response['choices'] && response['choices'][0] && response['choices'][0]['message']
+                answer = response['choices'][0]['message']['content']
+            else
+                raise "Invalid API response format"
             end
-            con += 1
+        rescue => e
+            # 오류 메시지 출력
+            puts "Error occurred: #{e.message}"
+            answer = "오류가 발생했습니다."
         end
 
-        print('api return ==> ')
-        puts(answer)
+        # "생성 중..." 메시지 출력 종료
+        thread.kill
 
+        # 결과 로그 출력
+        puts "Final API response ==> #{answer}"
         return answer
     end
+
+    def calculate_tokens(text)
+        # 간단한 방식으로 텍스트의 토큰 수 계산 (정확도는 다를 수 있음)
+        # OpenAI API는 1토큰이 대략 4글자 정도임
+        text.split(/\s+/).length  # 간단한 단어 수로 계산
+    end
 end
 
+
+
+
+
 class Chat_title
-    def initialize(api_key)
+    def initialize(api_key, gpt_title_prompt)
         @api_key = api_key
+        @gpt_title_prompt = gpt_title_prompt
     end
 
     def message(title)
+        puts 'Sending request to GPT...(제목 생성 중...)'
+        # "키워드 기반 글 생성 중..." 메시지를 별도 스레드로 처리
+        thread = Thread.new do
+            while true
+                print "▶"
+                sleep(3)
+            end
+        end
         url = 'https://api.openai.com/v1/chat/completions'
         headers = {
             'Content-Type' => 'application/json',
@@ -122,15 +136,15 @@ class Chat_title
             },
             {
                 "role" => "user",
-                "content" => "#{title}\n위 문장을 비슷한 길이로 ChatGPT의 멘트는 빼고 표현을 더 추가해서 하나만 만들어줘."
+                "content" => "#{@gpt_title_prompt}\n#{title}"
             }]
         }
 
         begin
             req = HTTP.headers(headers).post(url, json: data)
-            puts "HTTP Status: #{req.status}" # 상태 코드 확인
+           
             response = JSON.parse(req.body.to_s)
-            puts "API Response: #{response}" # 전체 응답 출력
+           
 
             if req.status == 429
                 return "API 요청 제한을 초과했습니다. 플랜 및 할당량을 확인하세요."
@@ -138,11 +152,18 @@ class Chat_title
 
             # 응답 데이터에서 안전하게 값 추출
             answer = response.dig('choices', 0, 'message', 'content')
-            answer ||= (title) # 응답이 없을 경우 기본 메시지 설정
+
+            # 따옴표 제거
+            answer = answer.gsub('"', '') if answer
+
+            answer ||= title # 응답이 없을 경우 기본 메시지 설정
         rescue => e
             puts "Error: #{e.message}"
             answer = "오류가 발생했습니다."
         end
+        
+        # "생성 중..." 메시지 출력 종료
+        thread.kill
 
         puts 'API return ==> '
         puts answer
@@ -150,12 +171,23 @@ class Chat_title
     end
 end
 
+
 class Chat_content
-    def initialize(api_key)
+    def initialize(api_key, gpt_content_prompt)
         @api_key = api_key
+        @gpt_content_prompt = gpt_content_prompt
     end
 
     def message(content)
+        puts 'Sending request to GPT...(내용 변형 중...)'
+        puts '주의:GPT 특성상 원고 길이가 공백 포함 4천자를 넘기면 오류가 발생할 수 있습니다.'
+        # "키워드 기반 글 생성 중..." 메시지를 별도 스레드로 처리
+        thread = Thread.new do
+            while true
+                print "▶"
+                sleep(3)
+            end
+        end
     
         url = 'https://api.openai.com/v1/chat/completions'
         headers = {
@@ -170,15 +202,16 @@ class Chat_content
             },
             {
                 "role" => "user",
-                "content" => "#{content}\nChatGPT의 멘트는 빼고 위 전체적인 내용의 형식을 똑같이 표현을 더 추가하고 유사어로 변경하여 하나 만들어줘! 전화번호,연락처,가격,홈페이지안내 ,상담안내 관련 문구는 유지해야해"
+                "content" => "#{@gpt_content_prompt}\n#{content}"
+              
             }]
         }
 
         begin
             req = HTTP.headers(headers).post(url, json: data)
-            puts "HTTP Status: #{req.status}" # 상태 코드 확인
+            
             response = JSON.parse(req.body.to_s)
-            puts "API Response: #{response}" # 전체 응답 출력
+            
             
             if req.status == 429
                 return "API 요청 제한을 초과했습니다. 플랜 및 할당량을 확인하세요."
@@ -192,6 +225,9 @@ class Chat_content
             answer = "오류가 발생했습니다."
         end
 
+        # "생성 중..." 메시지 출력 종료
+        thread.kill
+
         puts 'API return ==> '
         puts answer
         answer
@@ -199,6 +235,7 @@ class Chat_content
 end
 
 
+
 #############################################gpt############################################
 
 class Naver
@@ -297,6 +334,7 @@ class Naver
 
 
 
+
     def login(user_id, user_pw, proxy)
         @user_id = user_id
         @user_id11 = user_id
@@ -401,7 +439,7 @@ class Naver
                 end
             end
             return 0 
-            @driver.quit
+            @driver.quit 
         end  
     end
 
@@ -701,6 +739,7 @@ class Naver
             rescue 
             end
 
+            
         begin    
         #제목 입력
         wait = Selenium::WebDriver::Wait.new(:timeout => 10)
@@ -717,9 +756,9 @@ class Naver
                 puts "Failed to close tab: #{e.message}"
                 end
             end
-            return 0  
-            @driver.quit
-        end      
+            return 0 
+            @driver.quit 
+        end 
 
         ele = @driver.find_element(:xpath, '//*[@draggable="false"]')
         @driver.action.click(ele).perform
@@ -1626,10 +1665,10 @@ class Naver
             return 0  
             @driver.quit
         end  
-        
 
 
-        
+
+   
         category = option['category']
         begin
             p category
@@ -1715,6 +1754,7 @@ class Naver
    
         # puts data['documentModel'] = data['documentModel'].to_json
 
+
         if option['발행'] == '2' #저장
             sleep(dd_time.to_i)
             @driver.find_element(:xpath, '//*[@class="save_btn__bzc5B"]').click
@@ -1791,7 +1831,7 @@ class Naver
 
         end
 
-
+        
      
     end
 end
@@ -2440,6 +2480,7 @@ class Wordpress
                                 puts 'adb error pass'
                             end
                         end
+                           
 
                         check_success = 1
                         @data['table'][index][-1] =  0
@@ -2458,10 +2499,19 @@ class Wordpress
                         end
 
                         if @data['포스트설정']['gpt제목'].checked?
-                            chat = Chat_title.new(@data['포스트설정']['api_key'].text.to_s.force_encoding('utf-8'))
+                            gpt_title_prompt = @data['포스트설정']['gpt제목_프롬프트'].text.to_s.force_encoding('utf-8')
+                        
+                            # 공백을 포함한 빈 문자열을 체크하기 위해 strip을 사용
+                            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)
+                        
+                            # 메시지 요청 후 title에 저장
                             gpt_text1 = chat.message(title)
                             title = gpt_text1.to_s
                         end
+                        
                     
 
                         @data['table'][index][-1] =  5
@@ -2504,18 +2554,16 @@ class Wordpress
 
                         
                         if @data['포스트설정']['gpt내용'].checked?
+                            gpt_content_prompt = @data['포스트설정']['gpt내용_프롬프트'].text.to_s.force_encoding('utf-8')
                             api_key = @data['포스트설정']['api_key'].text.to_s.force_encoding('utf-8')
-                            #key_change = @data['포스트설정']['특정단어키워드로변경값'].text.to_s.force_encoding('utf-8')
-                            #imotcon_change = @data['포스트설정']['스티커로변경단어'].text.to_s.force_encoding('utf-8')
-                            #template_change = @data['포스트설정']['내템플릿변경단어'].text.to_s.force_encoding('utf-8')
-                            #ttdanar_change = @data['포스트설정']['단어링크적용단어'].text.to_s.force_encoding('utf-8')
-                            #sajine_change = @data['포스트설정']['단어사진으로변경단어'].text.to_s.force_encoding('utf-8')
-                            #mov_change = @data['포스트설정']['영상으로변경단어'].text.to_s.force_encoding('utf-8')
-                            #map_change = @data['포스트설정']['지도로변경단어'].text.to_s.force_encoding('utf-8')
-                            #inyong9_change = @data['포스트설정']['인용구변경단어'].text.to_s.force_encoding('utf-8')
-
                         
-                            chat = Chat_content.new(api_key)
+                            # 공백을 포함한 빈 문자열을 체크하기 위해 strip을 사용
+                            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)
+                        
+                            # 메시지 요청 후 content에 저장
                             gpt_text3 = chat.message(content)
                             content = gpt_text3.to_s
                         end
@@ -2719,13 +2767,17 @@ class Wordpress
                         @data['table'] << []
                         @data['table'].pop
                         if @data['포스트설정']['gpt키워드'].checked?
-                            chat = Chat.new(@data['포스트설정']['api_key'].text.to_s.force_encoding('utf-8'))
+                            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)
                             gpt_text = chat.message(keyword)
-                            content = content.to_s + "\n(자동생성글)\n" + gpt_text.to_s
+                            #content = content.to_s + "\n(자동생성글)\n" + gpt_text.to_s
+                            content = content.to_s + "(자동생성글)" + gpt_text.to_s
                         elsif @data['포스트설정']['내용을자동생성'].checked?
                             content = auto_text
                         elsif @data['포스트설정']['내용과자동생성'].checked?
-                            content = content + "\n(자동생성글)\n" + auto_text
+                            #content = content + "\n(자동생성글)\n" + auto_text
+                            content = content + "(자동생성글)" + auto_text
                         end
 
                         if @data['포스트설정']['내용키워드삽입'].checked?
@@ -2760,7 +2812,8 @@ class Wordpress
                             end
 
                             if @data['포스트설정']['내용을자동생성'].checked?
-                                content2 = content.split("\n")
+                                #content2 = content.split("\n")
+                                content2 = content.split
                             end
 
                             if @data['포스트설정']['내용과자동생성'].checked? or @data['포스트설정']['gpt키워드'].checked?
@@ -4081,6 +4134,11 @@ class Wordpress
                                 }
                             }  
                         } } 
+                        horizontal_box{
+                            stretchy false
+                                @data['포스트설정']['gpt키워드_프롬프트'] = entry(){
+                                text '프롬프트:관련 글을 1500자에서 2500자 사이로 만들어줘'
+                            }}    
                         horizontal_box{
                             stretchy false
                             grid{
@@ -4197,6 +4255,11 @@ class Wordpress
                               
                                   
                                 }}}
+                        horizontal_box{
+                            stretchy false
+                                @data['포스트설정']['gpt제목_프롬프트'] = entry(){
+                                text '프롬프트:비슷한 길이로 제목으로 사용할수있게 하나만 만들어줘'
+                            }}          
                         horizontal_box{
                             stretchy false
                             grid{
@@ -4310,6 +4373,11 @@ class Wordpress
                                                          
                                   
                                 }}}
+                        horizontal_box{
+                            stretchy false
+                                @data['포스트설정']['gpt내용_프롬프트'] = entry(){
+                                text '프롬프트:동의어,유사어를 이용해 변경해줘'
+                            }}        
                         horizontal_box{
                             stretchy false
                             grid{
@@ -4919,7 +4987,7 @@ class Wordpress
                                 left 1
                                 text 'URL'
                             }
-                            @data['포스트설정']['내용을자동생성'] = checkbox('키워드기반 생성으로만 등록(GPT 미 사용시 자체 생성)'){
+                            @data['포스트설정']['내용을자동생성'] = checkbox('키워드기반 글만 등록(GPT사용시 체크 해제)'){
                                 top 9
                                 left 0
                                 on_toggled{
@@ -4931,10 +4999,19 @@ class Wordpress
                                     end
                                 }
                             }
-
-                        
+                            label('※GPT사용시 내용설정 탭에서 설정'){
+                                top 9
+                                left 1
+                            }
+                          
+                            label('※GPT 미 사용시 세팅 권장'){
+                                top 9
+                                left 3
+                            }
+                            
+                       
                             aa1 = 2
-                            @data['포스트설정']['내용과자동생성'] = checkbox('내용파일+키워드기반 생성 등록(GPT 미 사용시 자체 생성)') {
+                            @data['포스트설정']['내용과자동생성'] = checkbox('원고+키워드기반 글 등록(GPT사용시 체크 해제)') {
                                 top 10 + aa1
                                 left 0
                                 on_toggled {
@@ -4950,7 +5027,14 @@ class Wordpress
                                   end
                                 }
                               }
-                              
+                              label('※GPT사용시 내용설정 탭에서 설정'){
+                                top 10 + aa1
+                                left 1
+                              }
+                              label('※GPT 미 사용시 세팅 권장'){
+                                top 10 + aa1
+                                left 3
+                            }
                               @data['포스트설정']['내용투명'] = checkbox('키워드 기반 자동 생성글 안보이게 처리') {
                                 top 11 + aa1
                                 left 0

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: nblog_duo
 version: !ruby/object:Gem::Version
-  version: 0.0.77
+  version: 0.0.90
 platform: ruby
 authors:
 - zon
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-02-12 00:00:00.000000000 Z
+date: 2025-02-17 00:00:00.000000000 Z
 dependencies: []
 description: File to Clipboard gem
 email: mymin26@naver.com