Update app.py

app.py

@@ -6,6 +6,7 @@ import tempfile
 import os
 from pathlib import Path
 import re
+import shutil
 
 def clean_json_content(content: str) -> str:
     """JSON 문자열에서 주석을 제거하고 JSON 형식을 정리합니다."""
@@ -65,18 +66,14 @@ def process_json_files(files: List[tempfile._TemporaryFileWrapper]) -> tuple[str
         df = pd.DataFrame(all_data)
         df_deduplicated = df.drop_duplicates(subset=['repo'])
         
-        # 임시 디렉토리 생성
-        temp_dir = Path(tempfile.gettempdir()) / "json_dedup"
-        temp_dir.mkdir(parents=True, exist_ok=True)
-        
-        # 결과를 임시 JSON 파일로 저장
-        output_path = temp_dir / "deduplicated_data.json"
+        # 결과를 loras.json 파일로 저장
+        output_path = "loras.json"
         result_json = df_deduplicated.to_dict('records')
         
         with open(output_path, 'w', encoding='utf-8') as f:
             json.dump(result_json, f, ensure_ascii=False, indent=2)
         
-        return str(output_path), f"성공적으로 처리되었습니다. 중복 제거 전 {len(all_data)}개, 중복 제거 후 {len(df_deduplicated)}개의 항목이 있습니다."
+        return output_path, f"성공적으로 처리되었습니다. 중복 제거 전 {len(all_data)}개, 중복 제거 후 {len(df_deduplicated)}개의 항목이 있습니다."
         
     except Exception as e:
         return None, f"처리 중 오류가 발생했습니다: {str(e)}"
@@ -99,25 +96,23 @@ def create_example_file():
         }
     ]
     
-    temp_dir = Path(tempfile.gettempdir()) / "json_dedup"
-    temp_dir.mkdir(parents=True, exist_ok=True)
-    example_path = temp_dir / "example.json"
+    example_path = "example_loras.json"
     
     with open(example_path, 'w', encoding='utf-8') as f:
         json.dump(example_data, f, ensure_ascii=False, indent=2)
     
-    return str(example_path)
+    return example_path
 
 # Gradio 인터페이스 생성
 iface = gr.Interface(
     fn=process_json_files,
     inputs=gr.File(file_count="multiple", label="JSON 파일 업로드 (여러 개 가능)"),
     outputs=[
-        gr.File(label="처리된 JSON 파일 다운로드"),
+        gr.File(label="loras.json 다운로드"),
         gr.Textbox(label="처리 결과")
     ],
     title="JSON 파일 중복 제거 도구",
-    description="repo 값을 기준으로 중복을 제거한 JSON 파일을 생성합니다.\n주석이나 불완전한 JSON 형식도 처리 가능합니다.",
+    description="repo 값을 기준으로 중복을 제거한 loras.json 파일을 생성합니다.\n주석이나 불완전한 JSON 형식도 처리 가능합니다.",
     examples=[[create_example_file()]]
 )