Update app.py

app.py

@@ -1,35 +1,43 @@
-import streamlit as st
 import os
-from git import Repo
 import sys
+from git import Repo
+import importlib.util
+
+def main():
+    # GitHub Repository and Directory Setup
+    repo_url = os.getenv("REPO_URL")  # The private repository URL from environment variables
+    github_token = os.getenv("GITHUB_PAT")  # GitHub Personal Access Token from environment variables
+    local_dir = "repo"
 
-# GitHub Reposi tory and Directory Setup
-repo_url = st.secrets["REPO_URL"]  # The private repository URL
-github_token = st.secrets["GITHUB_PAT"]  # GitHub Personal Access Token
-local_dir = "repo"
+    # Clone the private repository if it doesn't exist locally
+    if not os.path.exists(local_dir):
+        try:
+            # Prepare the repo URL with the token for authentication
+            repo_url = repo_url.replace("https://", f"https://{github_token}@")
+            print("Cloning the repository...")
+            Repo.clone_from(repo_url, local_dir)
+            print("Repository cloned successfully!")
+            
+            # Add the repo directory to the system path
+            sys.path.insert(0, local_dir)
+        except Exception as e:
+            print(f"Error cloning repository: {e}")
+            sys.exit(1)  # Exit the script if the repo can't be cloned
 
-# Clone the private repository if it doesn't exist locally
-if not os.path.exists(local_dir):
+    # Dynamically import and execute the app.py from the cloned repository
     try:
-        repo_url = repo_url.replace("https://", f"https://{github_token}@")
-        st.write("Cloning the repository...")
-        Repo.clone_from(repo_url, local_dir)
-        st.success("Repository cloned successfully!")
-        
-        # Add the repo directory to the system path
-        sys.path.insert(0, local_dir)
-    except Exception as e:
-        st.error(f"Error cloning repository: {e}")
-        st.stop()  # Stop the app if the repo can't be cloned
+        spec = importlib.util.spec_from_file_location("cloned_app", os.path.join(local_dir, "app.py"))
+        module = importlib.util.module_from_spec(spec)
+        spec.loader.exec_module(module)
 
-# Dynamically import and execute the app.py from the cloned repository
-try:
-    import importlib.util
-    spec = importlib.util.spec_from_file_location("cloned_app", os.path.join(local_dir, "app.py"))
-    module = importlib.util.module_from_spec(spec)
-    spec.loader.exec_module(module)
+        # If the module has a 'main' function, call it (for apps like FastAPI or Gradio)
+        if hasattr(module, "main"):
+            module.main()
+        else:
+            print("No main function found in the cloned app.")
+    except Exception as e:
+        print(f"Error running the app from the cloned repo: {e}")
+        sys.exit(1)  # Exit the script if the cloned app can't be executed
 
-except Exception as e:
-    st.error(f"Error running the app from the cloned repo: {e}")
-    st.stop()  # Stop the app if the cloned app can't be executed
-    st.error(f"Failed to process video: {str(e)}")
+if __name__ == "__main__":
+    main()