Update app-backup1.py

app-backup1.py

@@ -173,6 +173,130 @@ async def try_openai_api(openai_messages):
         raise e
 
 
+# Import to Package Name Mapping Dictionary
+IMPORT_TO_PACKAGE = {
+    'PIL': 'pillow',
+    'cv2': 'opencv-python',
+    'sklearn': 'scikit-learn',
+    'bs4': 'beautifulsoup4',
+    'yaml': 'pyyaml',
+    'tensorflow': 'tensorflow-cpu',
+    'tf': 'tensorflow-cpu',
+    'wx': 'wxPython',
+    'cairo': 'pycairo',
+    'MySQLdb': 'mysqlclient',
+    'psycopg2': 'psycopg2-binary',
+    'magic': 'python-magic',
+    'Image': 'pillow',
+    'skimage': 'scikit-image',
+    'matplotlib.pyplot': 'matplotlib',
+    'plt': 'matplotlib',
+    'np': 'numpy',
+    'pd': 'pandas',
+    'torch': 'pytorch',
+    'transformers': 'transformers',
+    'librosa': 'librosa',
+    'nltk': 'nltk',
+    'spacy': 'spacy',
+    'requests': 'requests',
+    'flask': 'flask',
+    'fastapi': 'fastapi',
+    'aiohttp': 'aiohttp',
+    'imageio': 'imageio',
+    'moviepy': 'moviepy',
+    'networkx': 'networkx',
+    'statsmodels': 'statsmodels',
+    'seaborn': 'seaborn',
+    'bokeh': 'bokeh',
+    'plotly': 'plotly',
+    'dash': 'dash',
+    'streamlit': 'streamlit',
+    'altair': 'altair',
+    'geopandas': 'geopandas',
+    'folium': 'folium',
+    'shapely': 'shapely',
+    'fiona': 'fiona',
+    'rasterio': 'rasterio',
+    'gdal': 'gdal',
+    'pyproj': 'pyproj',
+    'rtree': 'rtree',
+    'cartopy': 'cartopy',
+    'geoplot': 'geoplot',
+    'descartes': 'descartes',
+    'pysal': 'pysal',
+    'geopy': 'geopy',
+    'osmnx': 'osmnx',
+    'contextily': 'contextily',
+    'xarray': 'xarray',
+    'netCDF4': 'netCDF4',
+    'h5py': 'h5py',
+    'tables': 'pytables',
+    'zarr': 'zarr',
+    'dask': 'dask',
+    'numba': 'numba',
+    'sympy': 'sympy',
+    'scipy': 'scipy',
+    'scikit-image': 'scikit-image',
+    'scikit-learn': 'scikit-learn',
+    'keras': 'keras',
+    'theano': 'theano',
+    'caffe': 'caffe',
+    'mxnet': 'mxnet',
+    'chainer': 'chainer',
+    'pytorch': 'torch',
+    'tensorflow-gpu': 'tensorflow-gpu',
+    'cupy': 'cupy',
+    'pycuda': 'pycuda',
+    'pyopencl': 'pyopencl',
+    'pyvista': 'pyvista',
+    'mayavi': 'mayavi',
+    'vtk': 'vtk',
+    'trimesh': 'trimesh',
+    'open3d': 'open3d-python',
+    'pyqt5': 'PyQt5',
+    'pyside2': 'PySide2',
+    'tkinter': 'tk',
+    'kivy': 'kivy',
+    'pygame': 'pygame',
+    'arcade': 'arcade',
+    'pyglet': 'pyglet',
+    'panda3d': 'panda3d',
+    'ursina': 'ursina',
+    'moderngl': 'moderngl',
+    'glfw': 'glfw',
+    'pyopengl': 'PyOpenGL',
+    'pysdl2': 'PySDL2',
+    'pybullet': 'pybullet',
+    'box2d': 'box2d-py',
+    'pymunk': 'pymunk',
+    'pyode': 'pyode',
+    'pyrr': 'pyrr',
+    'noise': 'noise',
+    'wave': 'wave',
+    'sounddevice': 'sounddevice',
+    'pyaudio': 'PyAudio',
+    'simpleaudio': 'simpleaudio',
+    'pygame.mixer': 'pygame',
+    'pydub': 'pydub',
+    'aubio': 'aubio',
+    'music21': 'music21',
+    'pretty_midi': 'pretty_midi',
+    'mido': 'mido',
+    'fluidsynth': 'fluidsynth',
+    'mingus': 'mingus',
+    'pyfluidsynth': 'pyfluidsynth',
+    'python-rtmidi': 'python-rtmidi',
+    'pygame.midi': 'pygame',
+    'soundfile': 'soundfile',
+    'resampy': 'resampy'
+}
+
+def get_package_name(import_name):
+    """임포트명으로부터 실제 패키지명을 반환"""
+    # 점이 있는 경우 첫 부분만 사용 (예: matplotlib.pyplot -> matplotlib)
+    base_import = import_name.split('.')[0]
+    return IMPORT_TO_PACKAGE.get(base_import, base_import)
+
 def analyze_code(code: str) -> str:
     """코드 분석 결과를 HTML 형식으로 반환"""
     analysis = []
@@ -195,13 +319,14 @@ def analyze_code(code: str) -> str:
     for line in code.split('\n'):
         if line.startswith('import ') or line.startswith('from '):
             imports.append(line.strip())
-            # 패키지 이름 추출
+            # 패키지 이름 추출 및 변환
             if line.startswith('import '):
                 package = line.split('import ')[1].split()[0].split('.')[0]
             else:
                 package = line.split('from ')[1].split()[0].split('.')[0]
-            required_packages.add(package)
+            required_packages.add(get_package_name(package))
     
+   
     if imports:
         analysis.append("<h2>📚 필요한 라이브러리</h2>")
         analysis.append("<ul>")
@@ -473,109 +598,44 @@ def load_json_data():
     return [
         # 초급 레벨 (5건) - 기본 문법과 UI 익히기
         {
-            "name": "[초급-1] 구구단 계산기",
+            "name": "[기본] 구구단 계산기",
             "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),             
             "prompt": "구구단을 계산하고 출력하는 간단한 계산기를 만드세요. 사용자가 숫자를 입력하면 해당 단의 구구단이 출력되고, '전체 구구단 보기' 버튼을 누르면 2~9단까지 모두 표시됩니다."
         },
         {
-            "name": "[초급-2] BMI 계산기",
+            "name": "[기본] BMI 계산기",
             "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
             "prompt": "키(cm)와 몸무게(kg)를 입력받아 BMI를 계산하고, 비만도 판정 결과를 시각적으로 표시하는 앱을 만드세요. 결과는 저체중/정상/과체중/비만으로 구분하여 표시합니다."
         },
         {
-            "name": "[초급-3] 할인 계산기",
+            "name": "[기본] 할인 계산기",
             "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
             "prompt": "상품의 원가와 할인율을 입력받아 할인가를 계산하는 앱을 만드세요. 할인율은 슬라이더로 조절 가능하며, 최종 가격과 절약 금액을 실시간으로 표시합니다."
         },
         {
-            "name": "[초급-4] 단위 변환기",
+            "name": "[기본] 단위 변환기",
             "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
             "prompt": "길이(m/cm/km), 무게(kg/g), 온도(섭씨/화씨) 등 기본적인 단위 변환기를 만드세요. 드롭다운으로 변환 단위를 선택하고 실시간으로 결과가 업데이트됩니다."
         },
         {
-            "name": "[초급-5] 랜덤 번호 생성기",
+            "name": "[기본] 랜덤 번호 생성기",
             "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
             "prompt": "로또 번호나 랜덤 비밀번호를 생성하는 앱을 만드세요. 사용자가 범위와 개수를 지정할 수 있으며, 생성된 번호는 정렬되어 표시됩니다."
         },
 
-        # 중급 레벨 (5건) - 데이터 처리와 시각화
-        {
-            "name": "[중급-1] 데이터 시각화 대시보드",
-            "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
-            "prompt": "CSV 파일을 업로드하여 기본 통계(평균, 중앙값, 표준편차)를 계산하고, matplotlib으로 히스토그램, 박스플롯, 산점도를 그리는 대시보드를 만드세요."
-        },
-        {
-            "name": "[중급-2] 주식 데이터 분석기",
-            "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
-            "prompt": "주식 데이터를 분석하여 이동평균선, RSI, MACD 등 기술적 지표를 계산하고 시각화하는 앱을 만드세요. plotly를 사용하여 인터랙티브한 차트를 구현합니다."
-        },
         {
-            "name": "[중급-3] 이미지 편집기",
+            "name": "[기본] 이미지 편집기",
             "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
             "prompt": "PIL과 numpy를 사용하여 이미지 필터(흑백, 세피아, 블러 등), 회전, 리사이즈 기능이 있는 기본적인 이미지 편집기를 만드세요."
         },
         {
-            "name": "[중급-4] 일기예보 대시보드",
+            "name": "[기본] 일기예보 대시보드",
             "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
             "prompt": "OpenWeatherMap API를 사용하여 도시별 날씨 정보를 가져오고, 온도, 습도, 풍속 등을 시각화하는 대시보드를 만드세요."
         },
-        {
-            "name": "[중급-5] 음성 파형 분석기",
-            "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
-            "prompt": "librosa를 사용하여 음성 파일을 업로드하고 파형, 스펙트로그램을 표시하며 기본적인 오디오 특성(주파수, 진폭)을 분석하는 앱을 만드세요."
-        },
-
-        # 고급 레벨 (10건) - 머신러닝/딥러닝 기초
-        {
-            "name": "[고급-1] 손글씨 숫자 인식기",
-            "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
-            "prompt": "MNIST 데이터셋으로 학습된 CNN 모델을 사용하여 손으로 그린 숫자를 인식하는 앱을 만드세요. 캔버스에 그림을 그리면 실시간으로 예측합니다."
-        },
-        {
-            "name": "[고급-2] 얼굴 감지기",
-            "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
-            "prompt": "OpenCV와 dlib을 사용하여 이미지에서 얼굴을 감지하고 랜드마크를 표시하는 앱을 만드세요. 나이와 성별도 예측합니다."
-        },
-        {
-            "name": "[고급-3] 텍스트 감성 분석기",
-            "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
-            "prompt": "NLTK와 scikit-learn을 사용하여 텍스트의 감성(긍정/부정)을 분석하고 점수를 매기는 앱을 만드세요."
-        },
-        {
-            "name": "[고급-4] 이미지 세그멘테이션",
-            "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
-            "prompt": "U-Net 모델을 사용하여 이미지의 객체를 분할하고 마스크를 생성하는 앱을 만드세요."
-        },
-        {
-            "name": "[고급-5] 시계열 예측기",
-            "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
-            "prompt": "Prophet을 사용하여 시계열 데이터를 분석하고 미래 값을 예측하는 앱을 만드세요. 계절성과 트렌드를 시각화합니다."
-        },
-        {
-            "name": "[고급-6] 객체 탐지기",
-            "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
-            "prompt": "YOLO 모델을 사용하여 이미지나 비디오에서 객체를 탐지하고 바운딩 박스를 그리는 앱을 만드세요."
-        },
-        {
-            "name": "[고급-7] 음성 인식기",
-            "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
-            "prompt": "DeepSpeech 모델을 사용하여 음성을 텍스트로 변환하는 앱을 만드세요. 실시간 음성 입력도 지원합니다."
-        },
-        {
-            "name": "[고급-8] 이미지 스타일 변환",
-            "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
-            "prompt": "CycleGAN을 사용하여 이미지의 스타일을 다른 화풍으로 변환하는 앱을 만드세요."
-        },
-        {
-            "name": "[고급-9] 포즈 추정기",
-            "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
-            "prompt": "PoseNet을 사용하여 이미지나 비디오에서 사람의 포즈를 추정하고 관절을 표시하는 앱을 만드세요."
-        },
-        {
-            "name": "[고급-10] 이상치 탐지기",
-            "image_url": "data:image/png;base64," + get_image_base64('mouse.png'),   
-            "prompt": "Isolation Forest를 사용하여 데이터셋에서 이상치를 탐지하고 시각화하는 앱을 만드세요."
-        },
+ 
+   
+ 
 
         # 응용 레벨 (20건) - 허깅페이스 모델 통합
         {
@@ -974,133 +1034,8 @@ def deploy_to_huggingface(code: str):
         except Exception as e:
             raise e
 
-        # 4) 코드 정리 및 import 분석
+        # 4) 코드 정리
         code = code.replace("```python", "").replace("```", "").strip()
-        
-        # Import 문 추출 및 requirements.txt 생성을 위한 매핑
-        import_mapping = {
-            'numpy': 'numpy',
-            'pandas': 'pandas',
-            'torch': 'torch',
-            'matplotlib': 'matplotlib',
-            'plotly': 'plotly',
-            'transformers': 'transformers',
-            'PIL': 'Pillow',
-            'cv2': 'opencv-python',
-            'sklearn': 'scikit-learn',
-            'tensorflow': 'tensorflow',
-            'scipy': 'scipy',
-            'librosa': 'librosa',
-            'soundfile': 'soundfile',
-            'nltk': 'nltk',
-            'spacy': 'spacy',
-            'gensim': 'gensim',
-            'keras': 'keras',
-            'seaborn': 'seaborn',
-            'bokeh': 'bokeh',
-            'requests': 'requests',
-            'beautifulsoup4': 'beautifulsoup4',
-            'scikit-image': 'scikit-image',
-            'opencv-python': 'opencv-python',
-            'tqdm': 'tqdm',
-            'streamlit': 'streamlit',
-            'dash': 'dash',
-            'flask': 'flask',
-            'django': 'django',
-            'sqlalchemy': 'sqlalchemy',
-            'pymongo': 'pymongo',
-            'redis': 'redis',
-            'psycopg2': 'psycopg2-binary',
-            'mysql-connector': 'mysql-connector-python',
-            'pytest': 'pytest',
-            'unittest': 'unittest2',
-            'sphinx': 'sphinx',
-            'jupyter': 'jupyter',
-            'ipython': 'ipython',
-            'fastapi': 'fastapi',
-            'uvicorn': 'uvicorn',
-            'aiohttp': 'aiohttp',
-            'websockets': 'websockets',
-            'pyyaml': 'pyyaml',
-            'json': 'json5',
-            'xml': 'lxml',
-            'html': 'html5lib',
-            'urllib3': 'urllib3',
-            'cryptography': 'cryptography',
-            'bcrypt': 'bcrypt',
-            'jwt': 'pyjwt',
-            'pillow': 'Pillow',
-            'imageio': 'imageio',
-            'moviepy': 'moviepy',
-            'ffmpeg': 'ffmpeg-python',
-            'pydub': 'pydub',
-            'wave': 'wave',
-            'pyaudio': 'pyaudio',
-            'sounddevice': 'sounddevice',
-            'pygame': 'pygame',
-            'kivy': 'kivy',
-            'pyqt5': 'PyQt5',
-            'tkinter': 'tk',
-            'wx': 'wxPython',
-            'pyside2': 'PySide2',
-            'deap': 'deap',
-            'gym': 'gym',
-            'stable-baselines3': 'stable-baselines3',
-            'optuna': 'optuna',
-            'hyperopt': 'hyperopt',
-            'ray': 'ray',
-            'dask': 'dask',
-            'vaex': 'vaex',
-            'modin': 'modin',
-            'cupy': 'cupy',
-            'jax': 'jax',
-            'numba': 'numba',
-            'cython': 'cython',
-            'sympy': 'sympy',
-            'statsmodels': 'statsmodels',
-            'prophet': 'prophet',
-            'lightgbm': 'lightgbm',
-            'xgboost': 'xgboost',
-            'catboost': 'catboost',
-            'shap': 'shap',
-            'lime': 'lime',
-            'eli5': 'eli5',
-            'yellowbrick': 'yellowbrick',
-            'altair': 'altair',
-            'plotnine': 'plotnine',
-            'folium': 'folium',
-            'geopandas': 'geopandas',
-            'shapely': 'shapely',
-            'rasterio': 'rasterio',
-            'networkx': 'networkx',
-            'graphviz': 'graphviz',
-            'pydot': 'pydot',
-            'pygraphviz': 'pygraphviz'
-        }
-        
-        required_packages = set()
-
-        # 코드에서 import 문 분석
-        import_pattern = r'^(?:from\s+(\S+)|import\s+([^,\s]+)(?:\s*,\s*([^,\s]+))*)'
-        
-        for line in code.split('\n'):
-            line = line.strip()
-            if line.startswith('import ') or line.startswith('from '):
-                matches = re.match(import_pattern, line)
-                if matches:
-                    if matches.group(1):  # from ... import ...
-                        package = matches.group(1).split('.')[0]
-                        if package in import_mapping:
-                            required_packages.add(import_mapping[package])
-                    else:  # import ...
-                        packages = [p.strip() for p in re.findall(r'[\w.]+', line[7:])]
-                        for package in packages:
-                            package = package.split('.')[0]
-                            if package in import_mapping:
-                                required_packages.add(import_mapping[package])
-
-        # gradio는 항상 포함
-        required_packages.add('gradio==5.5.0')
 
         # 5) 전체 애플리케이션 코드 생성
         if "demo.launch()" not in code:
@@ -1119,8 +1054,21 @@ def deploy_to_huggingface(code: str):
             repo_type="space"
         )
 
-        # requirements.txt 생성 및 업로드
-        requirements = '\n'.join(sorted(required_packages))
+        # 7) requirements.txt 생성 및 업로드
+        analysis_result = analyze_code(code)
+        requirements = ""
+
+        # HTML에서 requirements.txt 섹션 찾기
+        if "<h3>📋 Requirements.txt</h3>" in analysis_result:
+            start_idx = analysis_result.find("<pre>") + 5
+            end_idx = analysis_result.find("</pre>")
+            if start_idx > 4 and end_idx > 0:
+                requirements = analysis_result[start_idx:end_idx].strip()
+
+        # requirements가 비어있으면 기본값 설정
+        if not requirements:
+            requirements = 'gradio==5.5.0'
+
         with open("requirements.txt", "w") as f:
             f.write(requirements)
             
@@ -1131,7 +1079,7 @@ def deploy_to_huggingface(code: str):
             repo_type="space"
         )
 
-        # 7) 결과 반환
+        # 8) 결과 반환
         space_url = f"https://huggingface.co/spaces/{username}/{space_name}"
         return f'배포 완료! Private Space로 생성되었습니다. <a href="{space_url}" target="_blank" style="color: #1890ff; text-decoration: underline; cursor: pointer;">여기를 클릭하여 Space 열기</a>'
 
@@ -1201,7 +1149,6 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
 # UI 수정 부분 - antd.Col(span=24, md=8) 내부의 버튼 컨테이너에 배포 버튼 추가:
                         with antd.Flex(gap="small", justify="space-between"):
                             btn = antd.Button("Send", type="primary", size="large")
-                            execute_btn = antd.Button("Code 실행", type="default", size="large")
                             deploy_btn = antd.Button("실행하기", type="default", size="large")  # 추가
                             clear_btn = antd.Button("Clear", type="default", size="large")
 
@@ -1250,11 +1197,7 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                     return None, gr.update(active_key="empty")
 
 # 이벤트 핸들러들
-            execute_btn.click(
-                fn=execute_code,
-                inputs=[input],
-                outputs=[sandbox, state_tab]
-            )
+
 
             codeBtn.click(
                 lambda: gr.update(open=True),