nblog_duo 0.0.90 → 0.0.93

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bfd2b0809846f63aeb22cf4514f862ce929334aab682d344f8a42e8dd3f62fd3
-  data.tar.gz: 4973fefc22f326598c49f4235ad261f86d9fe16c31d4c1b8035b05728a010f19
+  metadata.gz: db31678c1e8c72469d43b2ec52267cc966218dafe7e31259fb181e9bf83f4137
+  data.tar.gz: ef8000e946326e95eda34866c239804ee164c3fe5b773a122d1d6ac856b73e85
 SHA512:
-  metadata.gz: 8178cfc9537488957b3c3046ce3ae7ef74a3e6c8b4016dee9b12cc7878fe84a5332a2cfce099873db9a542c43e3e77cbd2d20bed501a1a4c2029037777183be9
-  data.tar.gz: 8cba1035a567b52a8a55c2694f1f3391833710b235b507959ecb5ab6515e9afbadb0a387494b925b17d2e7e69fe45cd52d7a17fa5ef2ca5f82f2ec3269a198c4
+  metadata.gz: d84296cb6debacafe2f68eedd3578b97ca58f5fd57f18f9845a12625aef9d428d32e38d44ac150b5a1336bed3f1249a3a5c4d2faae984400df80bb5f2eb05de1
+  data.tar.gz: 6c87a7b19110136b8cc9def03d9c3871c5932920d0c88acae5f345f9633ed7774bc6c379a219dc1c165a5ffcb0f6296686b8db13ab3a7b8f412178e1c88d7315

data/lib/nblog_duo.rb

@@ -236,6 +236,246 @@ end
 
 
 
+#############################################gpt############################################
+
+require 'glimmer-dsl-libui'
+require 'selenium-webdriver'
+# require 'webdrivers'
+require 'iconv'
+require 'nokogiri'
+require 'http'
+require 'json'
+require 'down'
+require 'rmagick'
+require 'fileutils'
+require 'rest-client'
+require 'open3'
+require 'clipboard'
+require 'crack'
+require 'uri'
+require 'cgi'
+require 'digest'
+require 'auto_click'
+require 'rainbow/refinement'
+include AutoClickMethods
+using Rainbow
+include Glimmer
+
+class Chat
+    def initialize(api_key, gpt_keyword_prompt)
+        @api_key = api_key
+        @gpt_keyword_prompt = gpt_keyword_prompt
+    end
+
+    def message(keyword)
+        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',
+            'Authorization' => 'Bearer ' + @api_key
+        }
+
+        # 사용자로부터 받은 입력과 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}\n#{@gpt_keyword_prompt}"
+                }
+            ],
+            'max_tokens' => max_response_tokens  # 최대 응답 토큰 설정
+        }
+
+  
+
+        answer = ''
+        begin
+            req = HTTP.headers(headers).post(url, :json => data)
+
+            # 상태 코드 확인
+            if req.status != 200
+                raise "HTTP Error: #{req.status}, Response Body: #{req.body.to_s}"
+            end
+
+            # 응답 내용 출력 (디버깅용)
+            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
+        rescue => e
+            # 오류 메시지 출력
+            puts "Error occurred: #{e.message}"
+            answer = "오류가 발생했습니다."
+        end
+
+        # "생성 중..." 메시지 출력 종료
+        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, 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',
+            'Authorization' => 'Bearer ' + @api_key
+        }
+        data = {
+            'model' => 'gpt-4',
+            'messages' => [{
+                "role" => "system",
+                "content" => "너는 매우 친절하고 성의 있게 답변하는 AI 어시스턴트야."
+            },
+            {
+                "role" => "user",
+                "content" => "#{@gpt_title_prompt}\n#{title}"
+            }]
+        }
+
+        begin
+            req = HTTP.headers(headers).post(url, json: data)
+           
+            response = JSON.parse(req.body.to_s)
+           
+
+            if req.status == 429
+                return "API 요청 제한을 초과했습니다. 플랜 및 할당량을 확인하세요."
+            end
+
+            # 응답 데이터에서 안전하게 값 추출
+            answer = response.dig('choices', 0, 'message', 'content')
+
+            # 따옴표 제거
+            answer = answer.gsub('"', '') if answer
+
+            answer ||= title # 응답이 없을 경우 기본 메시지 설정
+        rescue => e
+            puts "Error: #{e.message}"
+            answer = "오류가 발생했습니다."
+        end
+        
+        # "생성 중..." 메시지 출력 종료
+        thread.kill
+
+        puts 'API return ==> '
+        puts answer
+        answer
+    end
+end
+
+
+class Chat_content
+    def initialize(api_key, gpt_content_prompt)
+        @api_key = api_key
+        @gpt_content_prompt = gpt_content_prompt
+    end
+
+    def message(content)
+        puts '주의:GPT 특성상 원고 길이가 공백 포함 4천자를 넘기면 오류가 발생할 수 있습니다.'
+        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',
+            'Authorization' => 'Bearer ' + @api_key
+        }
+        data = {
+            'model' => 'gpt-4',
+            'messages' => [{
+                "role" => "system",
+                "content" => "너는 매우 친절하고 성의 있게 답변하는 AI 어시스턴트야."
+            },
+            {
+                "role" => "user",
+                "content" => "#{@gpt_content_prompt}\n#{content}"
+              
+            }]
+        }
+
+        begin
+            req = HTTP.headers(headers).post(url, json: data)
+            
+            response = JSON.parse(req.body.to_s)
+            
+            
+            if req.status == 429
+                return "API 요청 제한을 초과했습니다. 플랜 및 할당량을 확인하세요."
+            end
+
+            # 응답 데이터에서 안전하게 값 추출
+            answer = response.dig('choices', 0, 'message', 'content')
+            answer ||= (content) # 응답이 없을 경우 기본 메시지 설정
+        rescue => e
+            puts "Error: #{e.message}"
+            answer = "오류가 발생했습니다."
+        end
+
+        # "생성 중..." 메시지 출력 종료
+        thread.kill
+
+        puts 'API return ==> '
+        puts answer
+        answer
+    end
+end
+
+
+
 #############################################gpt############################################
 
 class Naver
@@ -246,9 +486,13 @@ class Naver
         naver_cookie_dir = "C:/naver_cookie"
         FileUtils.mkdir_p(naver_cookie_dir) unless File.exist?(naver_cookie_dir)
         if proxy == ''
-        system(%{"C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe" https://naver.com/ --remote-debugging-port=9222 --user-data-dir=C:/naver_cookie/#{user_id}})
+     
+            system(%{"C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe" https://naver.com/ --remote-debugging-port=9222 --user-data-dir=C:/naver_cookie/#{user_id} --no-first-run --no-default-browser-check --disable-sync})
+
         else  
-            system(%{"C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe" "https://naver.com/" --remote-debugging-port=9222 --user-data-dir=C:/naver_cookie/#{user_id} --proxy-server=#{proxy.to_s.force_encoding('utf-8').to_s}}) 
+            
+            system(%{"C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe" https://naver.com/ --remote-debugging-port=9222 --user-data-dir=C:/naver_cookie/#{user_id}  --proxy-server=#{proxy.to_s.force_encoding('utf-8').to_s} --no-first-run --no-default-browser-check --disable-sync})
+
         end
     end
     def chrome_start(proxy, user_id)
@@ -260,6 +504,9 @@ class Naver
                 options = Selenium::WebDriver::Chrome::Options.new
                 options.add_argument('--no-first-run') # 자동 실행 시 나타나는 "첫 실행" 화면 방지
                 options.add_argument('--disable-extensions') # 확장 프로그램 초기화 화면 방지
+                options.add_argument('--disable-sync')  # Chrome 동기화 비활성화
+                options.add_argument('--disable-popup-blocking') # 팝업 방지 (로그인 창이 뜨는 경우 대비)
+                options.add_argument('--no-default-browser-check')
                 options.page_load_strategy = :normal
                 options.timeouts = {page_load: 20_000}
                 options.page_load_strategy = 'none'
@@ -285,6 +532,9 @@ class Naver
                 options = Selenium::WebDriver::Chrome::Options.new
                 options.add_argument('--no-first-run') # 자동 실행 시 나타나는 "첫 실행" 화면 방지
                 options.add_argument('--disable-extensions') # 확장 프로그램 초기화 화면 방지
+                options.add_argument('--disable-sync')  # Chrome 동기화 비활성화
+                options.add_argument('--disable-popup-blocking') # 팝업 방지 (로그인 창이 뜨는 경우 대비)
+                options.add_argument('--no-default-browser-check')
                 options.add_argument '--proxy-server='+proxy.to_s.force_encoding('utf-8').to_s
                 options.page_load_strategy = :normal
                 options.timeouts = {page_load: 20_000}
@@ -309,6 +559,9 @@ class Naver
                     options = Selenium::WebDriver::Chrome::Options.new
                     options.add_argument('--no-first-run') # 자동 실행 시 나타나는 "첫 실행" 화면 방지
                     options.add_argument('--disable-extensions') # 확장 프로그램 초기화 화면 방지
+                    options.add_argument('--disable-sync')  # Chrome 동기화 비활성화
+                    options.add_argument('--disable-popup-blocking') # 팝업 방지 (로그인 창이 뜨는 경우 대비)
+                    options.add_argument('--no-default-browser-check')
                     options.page_load_strategy = :normal
                     options.timeouts = {page_load: 20_000}
                     options.page_load_strategy = 'none'
@@ -373,13 +626,20 @@ class Naver
             puts'[Step.02] 계정 세션 확인!! 로그인 skip.......'.yellow
             sleep(2.5)
         rescue
-            wait = Selenium::WebDriver::Wait.new(:timeout => 7)
-            #요소가 나타날 때까지 3초 동안 기다립니다.
-            wait.until { @driver.find_element(:xpath, '//*[@class="MyView-module__link_login___HpHMW"]') }
-            sleep(1.5)
-            @driver.find_element(:xpath, '//*[@class="MyView-module__link_login___HpHMW"]').click
-            check_cookie_login = 0
-            sleep(1)
+            begin
+                wait = Selenium::WebDriver::Wait.new(:timeout => 7)
+                # 요소가 나타날 때까지 기다립니다.
+                wait.until { @driver.find_element(:xpath, '//*[@class="MyView-module__link_login___HpHMW"]') }
+                sleep(1.5)
+        
+                # 요소가 있으면 클릭
+                @driver.find_element(:xpath, '//*[@class="MyView-module__link_login___HpHMW"]').click
+                check_cookie_login = 0
+                sleep(1)
+            rescue
+                @driver.quit 
+                return 0     
+            end
         end
 
         if check_cookie_login == 0
@@ -414,8 +674,8 @@ class Naver
                     puts "Failed to close tab: #{e.message}"
                     end
                 end
-                return 0  
                 @driver.quit
+                return 0  
             end  
             
         else
@@ -438,9 +698,9 @@ class Naver
                 puts "Failed to close tab: #{e.message}"
                 end
             end
-            return 0 
             @driver.quit 
-        end  
+            return 0 
+            end  
     end
 
     def create_id
@@ -756,8 +1016,9 @@ class Naver
                 puts "Failed to close tab: #{e.message}"
                 end
             end
-            return 0 
             @driver.quit 
+            return 0 
+            
         end 
 
         ele = @driver.find_element(:xpath, '//*[@draggable="false"]')
@@ -1662,8 +1923,9 @@ class Naver
                 puts "Failed to close tab: #{e.message}"
                 end
             end
-            return 0  
             @driver.quit
+            return 0  
+            
         end  
 
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: nblog_duo
 version: !ruby/object:Gem::Version
-  version: 0.0.90
+  version: 0.0.93
 platform: ruby
 authors:
 - zon