Update main.py

main.py

@@ -5,6 +5,7 @@ from telethon.sessions import StringSession
 from fastapi import FastAPI, UploadFile, File, HTTPException
 from fastapi.responses import FileResponse
 import uvicorn
+import tempfile # Import the tempfile module
 
 # --- Configuration from environment variables ---
 # Your Telegram session string
@@ -17,10 +18,9 @@ API_HASH = os.environ.get('API_HASH')
 # The Telegram channel you want to upload files to
 TARGET_CHANNEL = 'https://t.me/TestNAI01'
 
-# Directory to store uploaded files temporarily
-UPLOAD_DIR = "uploads"
-# This line ensures the directory exists. It will now succeed due to Dockerfile permissions.
-os.makedirs(UPLOAD_DIR, exist_ok=True)
+# --- FIX: Use the system's temporary directory ---
+# This directory is guaranteed to be writable and avoids permission errors.
+UPLOAD_DIR = tempfile.gettempdir()
 
 app = FastAPI()
 
@@ -31,19 +31,19 @@ async def upload_to_telegram(file_path: str):
     if not all([SESSION_STRING, API_ID, API_HASH]):
         raise ValueError("SESSION_STRING, API_ID, and API_HASH must be set as environment variables.")
 
-    # Initialize the client without proxy settings
+    # Initialize the client
     async with TelegramClient(StringSession(SESSION_STRING), API_ID, API_HASH) as client:
         try:
             await client.send_file(TARGET_CHANNEL, file_path)
         except Exception as e:
-            # Provide a more specific error message if upload fails
             raise HTTPException(status_code=500, detail=f"Failed to upload to Telegram: {e}")
 
 @app.post("/v1/upload")
 async def upload_file(file: UploadFile = File(...)):
     """
-    Receives a file and uploads it to the Telegram channel.
+    Receives a file, saves it to the temp directory, and uploads it to the Telegram channel.
     """
+    # Create a secure path in the temporary directory
     file_location = os.path.join(UPLOAD_DIR, file.filename)
     try:
         # Save the uploaded file temporarily
@@ -61,18 +61,17 @@ async def upload_file(file: UploadFile = File(...)):
 @app.get("/v1/download/{filename}")
 async def download_file(filename: str):
     """
-    Serves a file for download from the local 'uploads' directory.
-    NOTE: This is a basic implementation. For a production system, you would need a
-    more robust method to retrieve files, perhaps by downloading them from Telegram
-    on-demand instead of storing them locally.
+    Serves a file for download.
+    NOTE: This endpoint is for demonstration. It looks for files in the temp
+    directory, but files uploaded in previous requests will have been deleted.
+    A production system would need a proper way to retrieve files from Telegram.
     """
     file_path = os.path.join(UPLOAD_DIR, filename)
     if os.path.exists(file_path):
         return FileResponse(file_path)
-    raise HTTPException(status_code=404, detail="File not found on this server.")
+    raise HTTPException(status_code=404, detail="File not found. Files are only stored temporarily during upload.")
 
 if __name__ == "__main__":
-    # This block allows running the script directly for local testing (without Docker)
     if not all([SESSION_STRING, API_ID, API_HASH]):
         print("FATAL ERROR: The environment variables SESSION_STRING, API_ID, and API_HASH must be set.")
     else: