Update app.py

app.py

@@ -1,16 +1,9 @@
 #app.py
 import gradio as gr
+from gradio_client import Client
 import os
-import sys
 import tempfile
 import importlib.util
-import re
-from bs4 import BeautifulSoup
-import requests
-from fpdf import FPDF
-import markdown2
-from PIL import Image
-import google.generativeai as genai
 
 def load_module_from_env(env_var_name, module_name):
     """환경변수에서 모듈 코드를 가져와서 동적으로 로드"""
@@ -18,111 +11,9 @@ def load_module_from_env(env_var_name, module_name):
     if not module_code:
         raise ValueError(f"환경변수 {env_var_name}가 설정되지 않았습니다.")
     
-    # 필요한 임포트문들을 모듈 코드 앞에 추가
-    required_imports = """
-import os
-import re
-import random
-import logging
-import requests
-from bs4 import BeautifulSoup
-from fpdf import FPDF
-import markdown2
-from PIL import Image
-import google.generativeai as genai
-from typing import List, Dict, Any, Optional
-import sys
-import tempfile
-import json
-
-# Google Generative AI 설정 및 MockClient 정의
-try:
-    import google.generativeai as genai
-    
-    # 완전한 MockClient 클래스 정의
-    class MockClient:
-        def __init__(self, *args, **kwargs):
-            self.models = MockModels()
-            
-        def generate_content(self, *args, **kwargs):
-            return MockResponse()
-            
-    class MockModels:
-        def list(self, *args, **kwargs):
-            return []
-            
-        def get(self, model_name):
-            return MockModel()
-            
-    class MockModel:
-        def __init__(self):
-            self.name = "mock-model"
-            
-        def generate_content(self, *args, **kwargs):
-            return MockResponse()
-            
-    class MockResponse:
-        def __init__(self):
-            self.text = "환경설정 문제로 모의 응답을 반환합니다."
-            self.candidates = [MockCandidate()]
-            
-    class MockCandidate:
-        def __init__(self):
-            self.content = MockContent()
-            
-    class MockContent:
-        def __init__(self):
-            self.parts = [MockPart()]
-            
-    class MockPart:
-        def __init__(self):
-            self.text = "환경설정 문제로 모의 응답을 반환합니다."
-    
-    # Client 클래스가 없는 경우에만 대체 처리
-    if not hasattr(genai, 'Client'):
-        genai.Client = MockClient
-        print("MockClient로 대체되었습니다.")
-    else:
-        print("Google Generative AI Client를 사용합니다.")
-        
-except ImportError as e:
-    print(f"Google Generative AI 라이브러리 임포트 오류: {e}")
-    
-    # 완전한 genai 모듈 mock
-    class MockGenAI:
-        def __init__(self):
-            self.Client = MockClient
-            
-        def configure(self, *args, **kwargs):
-            pass
-            
-        def list_models(self):
-            return []
-            
-        def GenerativeModel(self, model_name):
-            return MockModel()
-    
-    genai = MockGenAI()
-"""
-    
-    # 기존 import문들을 제거하고 새로운 import문 추가
-    lines = module_code.split('\n')
-    filtered_lines = []
-    for line in lines:
-        stripped = line.strip()
-        # import문이나 from문이 아닌 경우만 포함
-        if not (stripped.startswith('import ') or stripped.startswith('from ') or stripped.startswith('#')):
-            filtered_lines.append(line)
-        elif stripped.startswith('#') and not any(keyword in stripped.lower() for keyword in ['import', 'from']):
-            # 주석은 import 관련이 아닌 경우만 포함
-            filtered_lines.append(line)
-    
-    # 새로운 모듈 코드 구성
-    full_module_code = required_imports + '\n' + '\n'.join(filtered_lines)
-    
     # 임시 파일 생성
     with tempfile.NamedTemporaryFile(mode='w', suffix='.py', delete=False, encoding='utf-8') as f:
-        f.write(full_module_code)
+        f.write(module_code)
         temp_file_path = f.name
     
     try:
@@ -131,185 +22,31 @@ except ImportError as e:
         module = importlib.util.module_from_spec(spec)
         spec.loader.exec_module(module)
         return module
-    except Exception as e:
-        print(f"모듈 {module_name} 로드 중 오류 발생: {e}")
-        return None
     finally:
         # 임시 파일 삭제
         if os.path.exists(temp_file_path):
             os.unlink(temp_file_path)
 
-# 환경변수 존재 여부 먼저 확인
-env_vars_status = {
-    'INFO_BLOG_CODE': bool(os.environ.get('INFO_BLOG_CODE')),
-    'TRAVEL_BLOG_CODE': bool(os.environ.get('TRAVEL_BLOG_CODE')),
-    'PRODUCT_REVIEW_BLOG_CODE': bool(os.environ.get('PRODUCT_REVIEW_BLOG_CODE')),
-    'FEATURE_FOCUS_BLOG_CODE': bool(os.environ.get('FEATURE_FOCUS_BLOG_CODE')),
-    'VISIT_REVIEW_BLOG_CODE': bool(os.environ.get('VISIT_REVIEW_BLOG_CODE'))
-}
-
-print("환경변수 설정 상태:", env_vars_status)
-missing_vars = [var for var, exists in env_vars_status.items() if not exists]
-if missing_vars:
-    print(f"설정되지 않은 환경변수: {missing_vars}")
-
-# 환경변수에서 각 모듈 로드
-try:
-    info_blog_module = load_module_from_env('INFO_BLOG_CODE', 'info_blog')
-    print("✓ INFO_BLOG_CODE 모듈 로드 성공")
-except ValueError as e:
-    print(f"✗ INFO_BLOG_CODE 로드 실패: {e}")
-    info_blog_module = None
-
-try:
-    travel_blog_module = load_module_from_env('TRAVEL_BLOG_CODE', 'travel_blog')
-    print("✓ TRAVEL_BLOG_CODE 모듈 로드 성공")
-except ValueError as e:
-    print(f"✗ TRAVEL_BLOG_CODE 로드 실패: {e}")
-    travel_blog_module = None
-
-try:
-    product_review_blog_module = load_module_from_env('PRODUCT_REVIEW_BLOG_CODE', 'product_review_blog')
-    print("✓ PRODUCT_REVIEW_BLOG_CODE 모듈 로드 성공")
-except ValueError as e:
-    print(f"✗ PRODUCT_REVIEW_BLOG_CODE 로드 실패: {e}")
-    product_review_blog_module = None
-
-try:
-    feature_focus_blog_module = load_module_from_env('FEATURE_FOCUS_BLOG_CODE', 'feature_focus_blog')
-    print("✓ FEATURE_FOCUS_BLOG_CODE 모듈 로드 성공")
-except ValueError as e:
-    print(f"✗ FEATURE_FOCUS_BLOG_CODE 로드 실패: {e}")
-    feature_focus_blog_module = None
-
-try:
-    visit_review_blog_module = load_module_from_env('VISIT_REVIEW_BLOG_CODE', 'visit_review_blog')
-    print("✓ VISIT_REVIEW_BLOG_CODE 모듈 로드 성공")
-except ValueError as e:
-    print(f"✗ VISIT_REVIEW_BLOG_CODE 로드 실패: {e}")
-    visit_review_blog_module = None
-
-# 로드 실패한 모듈들을 더미로 대체
-class DummyModule:
-    def generate_outline_1(self, *args): return "해당 모듈의 환경변수가 설정되지 않았습니다."
-    def generate_blog_post_1(self, *args): return "해당 모듈의 환경변수가 설정되지 않았습니다."
-    def generate_outline_2(self, *args): return "해당 모듈의 환경변수가 설정되지 않았습니다."
-    def generate_blog_post_2(self, *args): return "해당 모듈의 환경변수가 설정되지 않았습니다."
-    def generate_outline_3(self, *args): return "해당 모듈의 환경변수가 설정되지 않았습니다."
-    def generate_blog_post_3(self, *args): return "해당 모듈의 환경변수가 설정되지 않았습니다."
-    def generate_outline_4(self, *args): return "해당 모듈의 환경변수가 설정되지 않았습니다."
-    def generate_blog_post_4(self, *args): return "해당 모듈의 환경변수가 설정되지 않았습니다."
-    def generate_outline_5(self, *args): return "해당 모듈의 환경변수가 설정되지 않았습니다."
-    def generate_blog_post_5(self, *args): return "해당 모듈의 환경변수가 설정되지 않았습니다."
-
-dummy = DummyModule()
-if not info_blog_module: info_blog_module = dummy
-if not travel_blog_module: travel_blog_module = dummy
-if not product_review_blog_module: product_review_blog_module = dummy
-if not feature_focus_blog_module: feature_focus_blog_module = dummy
-if not visit_review_blog_module: visit_review_blog_module = dummy
-
-# 환경변수에서 로드된 모듈들의 함수를 전역으로 할당 (실제 모듈의 함수명에 맞춰 수정)
-try:
-    # 각 모듈에서 실제로 존재하는 함수명들을 확인하고 할당
-    print("info_blog_module 함수들:", [attr for attr in dir(info_blog_module) if not attr.startswith('_')])
-    print("travel_blog_module 함수들:", [attr for attr in dir(travel_blog_module) if not attr.startswith('_')])
-    print("product_review_blog_module 함수들:", [attr for attr in dir(product_review_blog_module) if not attr.startswith('_')])
-    print("feature_focus_blog_module 함수들:", [attr for attr in dir(feature_focus_blog_module) if not attr.startswith('_')])
-    print("visit_review_blog_module 함수들:", [attr for attr in dir(visit_review_blog_module) if not attr.startswith('_')])
-    
-    # info_blog 모듈 함수 할당 (실제 함수명 확인 후 할당)
-    if hasattr(info_blog_module, 'generate_outline_1'):
-        generate_outline_1 = info_blog_module.generate_outline_1
-    elif hasattr(info_blog_module, 'generate_outline'):
-        generate_outline_1 = info_blog_module.generate_outline
-    else:
-        generate_outline_1 = lambda *args: "info_blog 모듈에서 아웃라인 생성 함수를 찾을 수 없습니다."
-    
-    if hasattr(info_blog_module, 'generate_blog_post_1'):
-        generate_blog_post_1 = info_blog_module.generate_blog_post_1
-    elif hasattr(info_blog_module, 'generate_blog_post'):
-        generate_blog_post_1 = info_blog_module.generate_blog_post
-    else:
-        generate_blog_post_1 = lambda *args: "info_blog 모듈에서 블로그 생성 함수를 찾을 수 없습니다."
-    
-    # travel_blog 모듈 함수 할당
-    if hasattr(travel_blog_module, 'generate_outline_2'):
-        generate_outline_2 = travel_blog_module.generate_outline_2
-    elif hasattr(travel_blog_module, 'generate_outline'):
-        generate_outline_2 = travel_blog_module.generate_outline
-    else:
-        generate_outline_2 = lambda *args: "travel_blog 모듈에서 아웃라인 생성 함수를 찾을 수 없습니다."
-    
-    if hasattr(travel_blog_module, 'generate_blog_post_2'):
-        generate_blog_post_2 = travel_blog_module.generate_blog_post_2
-    elif hasattr(travel_blog_module, 'generate_blog_post'):
-        generate_blog_post_2 = travel_blog_module.generate_blog_post
-    else:
-        generate_blog_post_2 = lambda *args: "travel_blog 모듈에서 블로그 생성 함수를 찾을 수 없습니다."
-    
-    # product_review_blog 모듈 함수 할당
-    if hasattr(product_review_blog_module, 'generate_outline_3'):
-        generate_outline_3 = product_review_blog_module.generate_outline_3
-    elif hasattr(product_review_blog_module, 'generate_outline'):
-        generate_outline_3 = product_review_blog_module.generate_outline
-    else:
-        generate_outline_3 = lambda *args: "product_review_blog 모듈에서 아웃라인 생성 함수를 찾을 수 없습니다."
-    
-    if hasattr(product_review_blog_module, 'generate_blog_post_3'):
-        generate_blog_post_3 = product_review_blog_module.generate_blog_post_3
-    elif hasattr(product_review_blog_module, 'generate_blog_post'):
-        generate_blog_post_3 = product_review_blog_module.generate_blog_post
-    else:
-        generate_blog_post_3 = lambda *args: "product_review_blog 모듈에서 블로그 생성 함수를 찾을 수 없습니다."
-    
-    # feature_focus_blog 모듈 함수 할당
-    if hasattr(feature_focus_blog_module, 'generate_outline_4'):
-        generate_outline_4 = feature_focus_blog_module.generate_outline_4
-    elif hasattr(feature_focus_blog_module, 'generate_outline'):
-        generate_outline_4 = feature_focus_blog_module.generate_outline
-    else:
-        generate_outline_4 = lambda *args: "feature_focus_blog 모듈에서 아웃라인 생성 함수를 찾을 수 없습니다."
-    
-    if hasattr(feature_focus_blog_module, 'generate_blog_post_4'):
-        generate_blog_post_4 = feature_focus_blog_module.generate_blog_post_4
-    elif hasattr(feature_focus_blog_module, 'generate_blog_post'):
-        generate_blog_post_4 = feature_focus_blog_module.generate_blog_post
-    else:
-        generate_blog_post_4 = lambda *args: "feature_focus_blog 모듈에서 블로그 생성 함수를 찾을 수 없습니다."
-    
-    # visit_review_blog 모듈 함수 할당
-    if hasattr(visit_review_blog_module, 'generate_outline_5'):
-        generate_outline_5 = visit_review_blog_module.generate_outline_5
-    elif hasattr(visit_review_blog_module, 'generate_outline'):
-        generate_outline_5 = visit_review_blog_module.generate_outline
-    else:
-        generate_outline_5 = lambda *args: "visit_review_blog 모듈에서 아웃라인 생성 함수를 찾을 수 없습니다."
-    
-    if hasattr(visit_review_blog_module, 'generate_blog_post_5'):
-        generate_blog_post_5 = visit_review_blog_module.generate_blog_post_5
-    elif hasattr(visit_review_blog_module, 'generate_blog_post'):
-        generate_blog_post_5 = visit_review_blog_module.generate_blog_post
-    else:
-        generate_blog_post_5 = lambda *args: "visit_review_blog 모듈에서 블로그 생성 함수를 찾을 수 없습니다."
-        
-    print("✓ 모든 함수 할당 완료")
-    
-except AttributeError as e:
-    print(f"함수 할당 에러: {e}")
-    # 더미 함수들
-    def generate_outline_1(*args): return "환경변수가 설정되지 않았습니다."
-    def generate_blog_post_1(*args): return "환경변수가 설정되지 않았습니다."
-    def generate_outline_2(*args): return "환경변수가 설정되지 않았습니다."
-    def generate_blog_post_2(*args): return "환경변수가 설정되지 않았습니다."
-    def generate_outline_3(*args): return "환경변수가 설정되지 않았습니다."
-    def generate_blog_post_3(*args): return "환경변수가 설정되지 않았습니다."
-    def generate_outline_4(*args): return "환경변수가 설정되지 않았습니다."
-    def generate_blog_post_4(*args): return "환경변수가 설정되지 않았습니다."
-    def generate_outline_5(*args): return "환경변수가 설정되지 않았습니다."
-    def generate_blog_post_5(*args): return "환경변수가 설정되지 않았습니다."
-
-# 원본과 동일한 API 함수 정의 (함수명 그대로 유지)
+# 환경변수에서 각 모듈 로드 (원본과 동일한 방식)
+info_blog = load_module_from_env('INFO_BLOG_CODE', 'info_blog')
+travel_blog = load_module_from_env('TRAVEL_BLOG_CODE', 'travel_blog')
+product_review_blog = load_module_from_env('PRODUCT_REVIEW_BLOG_CODE', 'product_review_blog')
+feature_focus_blog = load_module_from_env('FEATURE_FOCUS_BLOG_CODE', 'feature_focus_blog')
+visit_review_blog = load_module_from_env('VISIT_REVIEW_BLOG_CODE', 'visit_review_blog')
+
+# 환경변수에서 로드된 모듈에서 함수 가져오기 (원본과 동일)
+generate_outline_1 = info_blog.generate_outline_1
+generate_blog_post_1 = info_blog.generate_blog_post_1
+generate_outline_2 = travel_blog.generate_outline_2
+generate_blog_post_2 = travel_blog.generate_blog_post_2
+generate_outline_3 = product_review_blog.generate_outline_3
+generate_blog_post_3 = product_review_blog.generate_blog_post_3
+generate_outline_4 = feature_focus_blog.generate_outline_4
+generate_blog_post_4 = feature_focus_blog.generate_blog_post_4
+generate_outline_5 = visit_review_blog.generate_outline_5
+generate_blog_post_5 = visit_review_blog.generate_blog_post_5
+
+# 기존 API 함수 정의 (정보성)
 def generate_outline_1_api(category, style, ref1, ref2, ref3):
     result = generate_outline_1(category, style, ref1, ref2, ref3)
     return result
@@ -318,7 +55,7 @@ def generate_blog_post_1_api(category, style, ref1, ref2, ref3, outline_input):
     result = generate_blog_post_1(category, style, ref1, ref2, ref3, outline_input)
     return result
 
-# API 함수 정의 (여행)
+# 기존 API 함수 정의 (여행)
 def generate_outline_2_api(category, style, ref1, ref2, ref3):
     result = generate_outline_2(category, style, ref1, ref2, ref3)
     return result
@@ -327,7 +64,7 @@ def generate_blog_post_2_api(category, style, ref1, ref2, ref3, outline_input):
     result = generate_blog_post_2(category, style, ref1, ref2, ref3, outline_input)
     return result
 
-# API 함수 정의 (상품후기)
+# 기존 API 함수 정의 (상품후기)
 def generate_outline_3_api(category, style, ref1, ref2, ref3):
     result = generate_outline_3(category, style, ref1, ref2, ref3)
     return result
@@ -336,7 +73,7 @@ def generate_blog_post_3_api(category, style, ref1, ref2, ref3, outline_input):
     result = generate_blog_post_3(category, style, ref1, ref2, ref3, outline_input)
     return result
 
-# API 함수 정의 (핵심기능집중형)
+# 새로운 API 함수 정의 (핵심기능집중형)
 def generate_outline_4_api(category, style, ref1, ref2, ref3):
     result = generate_outline_4(category, style, ref1, ref2, ref3)
     return result
@@ -345,7 +82,7 @@ def generate_blog_post_4_api(category, style, ref1, ref2, ref3, selected_feature
     result = generate_blog_post_4(category, style, ref1, ref2, ref3, selected_feature)
     return result
 
-# API 함수 정의 (방문후기형)
+# 새로운 API 함수 정의 (방문후기형)
 def generate_outline_5_api(category, style, ref1, ref2, ref3, photo_recommendations):
     result = generate_outline_5(category, style, ref1, ref2, ref3, photo_recommendations)
     return result