Update app.py

app.py

@@ -12,6 +12,69 @@ from datetime import datetime
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+def env_to_cookies(env_content: str, output_file: str) -> None:
+    """Convert environment variable content back to cookie file"""
+    try:
+        # Extract content from env format
+        if '="' not in env_content:
+            raise ValueError("Invalid env content format")
+            
+        content = env_content.split('="', 1)[1].strip('"')
+        
+        # Replace escaped newlines with actual newlines
+        cookie_content = content.replace('\\n', '\n')
+        
+        # Write to cookie file
+        with open(output_file, 'w') as f:
+            f.write(cookie_content)
+            
+    except Exception as e:
+        raise ValueError(f"Error converting to cookie file: {str(e)}")
+
+def save_to_env_file(env_content: str, env_file: str = '.env') -> None:
+    """Save environment variable content to .env file"""
+    try:
+        with open(env_file, 'w') as f:
+            f.write(env_content)
+        #print(f"Successfully saved to {env_file}")
+    except Exception as e:
+        raise ValueError(f"Error saving to env file: {str(e)}")
+
+def env_to_cookies_from_env(output_file: str) -> None:
+    """Convert environment variable from .env file to cookie file"""
+    try:
+        load_dotenv()  # Load from .env file
+        env_content = os.getenv('COOKIES')
+        #print(f"Printing env content: \n{env_content}")
+        if not env_content:
+            raise ValueError("COOKIES not found in .env file")
+            
+        env_to_cookies(f'COOKIES="{env_content}"', output_file)
+    except Exception as e:
+        raise ValueError(f"Error converting to cookie file: {str(e)}")
+
+def get_cookies():
+    """Get cookies from environment variable"""
+    load_dotenv()
+    cookie_content = os.getenv('COOKIES')
+    #print(cookie_content)
+    if not cookie_content:
+        raise ValueError("COOKIES environment variable not set")
+    return cookie_content
+
+def create_temp_cookie_file():
+    """Create temporary cookie file from environment variable"""
+    temp_cookie = tempfile.NamedTemporaryFile(mode='w+', delete=False, suffix='.txt')
+    try:
+        cookie_content = get_cookies()
+        # Replace escaped newlines with actual newlines
+        cookie_content = cookie_content.replace('\\n', '\n')
+        temp_cookie.write()
+        temp_cookie.flush()
+        return Path(temp_cookie.name)
+    finally:
+        temp_cookie.close()
+
 app = FastAPI(
     title="GAMDL API",
     description="API for downloading Google Drive files using gamdl",
@@ -36,6 +99,8 @@ class DownloadResponse(BaseModel):
 
 @app.post("/download", response_model=DownloadResponse)
 async def download_file(request: DownloadRequest):
+    cookiefile = "cookies.txt"
+    env_to_cookies_from_env("cookies.txt")
     try:
         # Create a unique subdirectory for this download
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")