Update app.py

app.py

@@ -1,22 +1,35 @@
-# -*- coding: utf-8 -*-
-
-import gradio as gr
-from huggingface_hub import InferenceClient
-from gradio_client import Client
 import os
-import requests
-import asyncio
-import logging
-from concurrent.futures import ThreadPoolExecutor
+import sys
+import streamlit as st
+from tempfile import NamedTemporaryFile
 
-import ast  #추가 삽입, requirements: albumentations 추가
-script_repr = os.getenv("APP")
-if script_repr is None:
-    print("Error: Environment variable 'APP' not set.")
-    sys.exit(1)
+def main():
+    try:
+        # Get the code from secrets
+        code = os.environ.get("MAIN_CODE")
+        
+        if not code:
+            st.error("⚠️ The application code wasn't found in secrets. Please add the MAIN_CODE secret.")
+            return
+        
+        # Create a temporary Python file
+        with NamedTemporaryFile(suffix='.py', delete=False, mode='w') as tmp:
+            tmp.write(code)
+            tmp_path = tmp.name
+        
+        # Execute the code
+        exec(compile(code, tmp_path, 'exec'), globals())
+        
+        # Clean up the temporary file
+        try:
+            os.unlink(tmp_path)
+        except:
+            pass
+            
+    except Exception as e:
+        st.error(f"⚠️ Error loading or executing the application: {str(e)}")
+        import traceback
+        st.code(traceback.format_exc())
 
-try:
-    exec(script_repr)
-except Exception as e:
-    print(f"Error executing script: {e}")
-    sys.exit(1)
+if __name__ == "__main__":
+    main()