nblog_duo 0.0.67 → 0.0.79

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 32be681d6aa2c5ea7f576ee6a65a2743432ac7556eb7a689cbfeb91756f83c94
-  data.tar.gz: a568440813e460a2fd872a7d42add1edf779df54a009f0627ad1fbc6ef8e13a1
+  metadata.gz: 16bacc65bd2507a5cba7950a748fffb95a23b4d535365ab99838cd790677fe72
+  data.tar.gz: a2aa69e059efdb23d7755a6514b7f8141c2858ae91cdd9bd2cf082e794a7fd97
 SHA512:
-  metadata.gz: 690967f42eb6c90231410f570d4769b7a1855b631f4080691d970592846bfb826a28f154e01cda9e5c46a56241dc87197bcbe52f02015eccfcc2db24e9f8a040
-  data.tar.gz: 1bcb4daa76ea1d8a89bf392b00c824d50809fd984f16f408c746a2c6986190d9174d7e7e510191269ac695cdf6b14cd5dcc21ba2b32f593ae65331fa143db3e3
+  metadata.gz: '08dc667f9a51c2da10687b3ae12ee5f591c69c40a198eb20a4476a4cdca1bbd4734ec1f7f1a0ec4c58af8544154ccfd459c971a7b532d502494920963e0395f7'
+  data.tar.gz: 2657492071bef35b031a2c00aca02b9a5b838b6bbc6746ee615af9772e54b71e406a15270852ff26fa63415adc623f4af7f93abf7ccb2a3da268292e106c9016

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
@@ -2403,56 +2443,44 @@ class Wordpress
                           
                               # ADB 서버 초기화
                               puts 'adb kill-server'
-                              Open3.capture3('adb kill-server')
-                              sleep(2)
+                              Open3.capture3('./adb kill-server')
+                              sleep(3)
                           
                               # 다시 ADB 서버 실행
                               puts 'adb start-server'
-                              Open3.capture3('adb start-server')
-                              sleep(2)
+                              Open3.capture3('./adb start-server')
+                              sleep(3)
                           
                               # 데이터를 끄고 켜기
                               puts 'adb -s ' + device_id + ' shell svc data disable'
-                              stdout2, stderr2, status2 = Open3.capture3('./adb -s ' + device_id + ' shell svc data disable')
-                          
-                              if status2.success?
-                                sleep(3)
-                                puts 'adb -s ' + device_id + ' shell svc data enable'
-                                Open3.capture3('./adb -s ' + device_id + ' shell svc data enable')
-                                sleep(3)
-                                puts 'adb ok'
-                                sleep(8)
-                          
-                                # IP 변경을 확인하는 람다 함수
-                                robot_ip = lambda do
-                                  loop do  # 무한 루프
-                                    begin
-                                      http = HTTP.get('https://www.findip.kr/')
-                                      noko = Nokogiri::HTML(http.to_s)
-                                      new_ip = noko.xpath('/html/body/header/h2').text.strip
-                          
-                                      if new_ip != @my_ip
-                                        @my_ip = new_ip
-                                        puts "IP 변경됨: #{@my_ip}"
-                                        break  # IP가 변경되었으면 루프 종료
-                                      else
-                                        puts 'IP가 변경되지 않음. 재시도...'
-                                        sleep(3)  # 3초 후 재시도
-                                      end
-                                    rescue => e
-                                      puts "IP 확인 중 오류 발생: #{e.message}. 재시도 중..."
-                                      sleep(3)  # 3초 후 재시도
-                                    end
-                                  end
+                              stdout2, stderr2, status2 = Open3.capture3('./adb -s '+device_id+' shell svc data disable')
+                              sleep(3)
+                              puts 'adb -s ' + device_id + ' shell svc data enable'
+                              Open3.capture3('./adb -s '+device_id+' shell svc data enable')
+                              sleep(3)
+                              puts 'adb ok'
+                              sleep(8)
+
+                              robot_ip = lambda do
+                                http = HTTP.get('https://www.findip.kr/')
+                                noko = Nokogiri::HTML(http.to_s)
+                                if noko.xpath('/html/body/header/h2').text != @my_ip
+                                    @my_ip = noko.xpath('/html/body/header/h2').text
+                                    puts "IP 변경됨[ #{@my_ip} ]"
+                                else
+                                    puts @my_ip
+                                    puts '제시도...'
+                                    sleep(3)
+                                    robot_ip[]
                                 end
-                                robot_ip[]  # 최초 호출
-                              else
-                                puts 'Failed to disable data. Error: ' + stderr2
-                              end
+                            end
+                            robot_ip[]
+
                             else
-                              puts 'adb error, unable to get devices. Error: ' + stderr
+                                puts 'adb error pass'
                             end
-                          end
+                        end
+                           
 
                         check_success = 1
                         @data['table'][index][-1] =  0
@@ -2471,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
@@ -2517,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
@@ -2732,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?
@@ -2773,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?
@@ -4094,6 +4134,11 @@ class Wordpress
                                 }
                             }  
                         } } 
+                        horizontal_box{
+                            stretchy false
+                                @data['포스트설정']['gpt키워드_프롬프트'] = entry(){
+                                text '프롬프트:관련 글을 1500자에서 2500자 사이로 만들어줘'
+                            }}    
                         horizontal_box{
                             stretchy false
                             grid{
@@ -4210,6 +4255,11 @@ class Wordpress
                               
                                   
                                 }}}
+                        horizontal_box{
+                            stretchy false
+                                @data['포스트설정']['gpt제목_프롬프트'] = entry(){
+                                text '프롬프트:비슷한 길이로 제목으로 사용할수있게 하나만 만들어줘'
+                            }}          
                         horizontal_box{
                             stretchy false
                             grid{
@@ -4323,6 +4373,11 @@ class Wordpress
                                                          
                                   
                                 }}}
+                        horizontal_box{
+                            stretchy false
+                                @data['포스트설정']['gpt내용_프롬프트'] = entry(){
+                                text '프롬프트:동의어,유사어를 이용해 변경해줘'
+                            }}        
                         horizontal_box{
                             stretchy false
                             grid{
@@ -4932,7 +4987,7 @@ class Wordpress
                                 left 1
                                 text 'URL'
                             }
-                            @data['포스트설정']['내용을자동생성'] = checkbox('키워드기반 생성으로만 등록(GPT 미 사용시 자체 생성)'){
+                            @data['포스트설정']['내용을자동생성'] = checkbox('키워드기반 글만 등록(GPT사용시 체크 해제)'){
                                 top 9
                                 left 0
                                 on_toggled{
@@ -4944,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 {
@@ -4963,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.67
+  version: 0.0.79
 platform: ruby
 authors:
 - zon
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-02-12 00:00:00.000000000 Z
+date: 2025-02-13 00:00:00.000000000 Z
 dependencies: []
 description: File to Clipboard gem
 email: mymin26@naver.com