Update app.py

app.py

@@ -27,9 +27,124 @@ async def stream_log(file_path):
             else:
                 await asyncio.sleep(0.1)
 
+import os
+import subprocess
+import urllib.request
+from pathlib import Path
+
+def install_python_version(python_version: str):
+    python_bin = f"/usr/local/bin/python{python_version}"
+    pip_bin = f"/usr/local/bin/pip{python_version}"
+
+    # Check if Python is already installed
+    if os.path.isfile(python_bin):
+        return python_bin, pip_bin
+
+    # Download and install Python if not present
+    try:
+        url = f"https://www.python.org/ftp/python/{python_version}/Python-{python_version}.tgz"
+        tar_path = f"/tmp/Python-{python_version}.tgz"
+        
+        # Download Python source tarball
+        urllib.request.urlretrieve(url, tar_path)
+        
+        # Extract tarball
+        subprocess.run(["tar", "-xzf", tar_path, "-C", "/tmp"], check=True)
+
+        # Build and install Python
+        python_src_path = f"/tmp/Python-{python_version}"
+        subprocess.run([f"{python_src_path}/configure"], cwd=python_src_path, check=True)
+        subprocess.run(["make", "-j"], cwd=python_src_path, check=True)
+        subprocess.run(["make", "altinstall", f"EXTRA_CFLAGS='-I/usr/local/include'"], cwd=python_src_path, check=True)
+        
+        # Clean up
+        os.remove(tar_path)
+        shutil.rmtree(python_src_path)
+        
+        # Update alternatives (if necessary)
+        subprocess.run(["update-alternatives", "--install", "/usr/bin/python3", "python3", python_bin, "1"], check=True)
+        subprocess.run(["update-alternatives", "--install", "/usr/bin/pip", "pip", pip_bin, "1"], check=True)
+        
+    except Exception as e:
+        raise RuntimeError(f"Error installing Python {python_version}: {str(e)}")
+
+    return python_bin, pip_bin
+
+# @app.post("/download-dependencies")
+# async def download_dependencies(requirements_file: UploadFile = File(...)):
+#     try:
+#         with NamedTemporaryFile(delete=False) as tmp:
+#             tmp.write(requirements_file.file.read())
+#             tmp_path = tmp.name
+
+#         # Ensure the directories exist
+#         os.makedirs("/tmp/dependencies", exist_ok=True)
+
+#         log_file_path = "/tmp/dependencies/download.log"
+#         with open(log_file_path, "w") as log_file:
+#             # Download dependencies
+#             result = subprocess.run(
+#                 ["pip", "download", "-r", tmp_path, "-d", "/tmp/dependencies"],
+#                 # stdout=log_file,
+#                 # stderr=log_file
+#             )
+
+#         if result.returncode != 0:
+#             raise HTTPException(status_code=500, detail="Error downloading dependencies. See log file for details.")
+
+#         # Create a tar file
+#         tar_path = "/tmp/dependencies.tar.gz"
+#         with tarfile.open(tar_path, "w:gz") as tar:
+#             for root, _, files in os.walk("/tmp/dependencies"):
+#                 for file in files:
+#                     file_path = os.path.join(root, file)
+#                     tar.add(file_path, arcname=file)
+#             tar.add(log_file_path, arcname="download.log")
+
+#         # Get the file size
+#         file_size = os.path.getsize(tar_path)
+
+#         # Return the tar file
+#         return StreamingResponse(open(tar_path, "rb"), media_type="application/gzip", headers={
+#             "Content-Disposition": f"attachment; filename=dependencies.tar.gz",
+#             "Content-Length": str(file_size)
+#         })
+
+#     except subprocess.CalledProcessError as e:
+#         raise HTTPException(status_code=500, detail=f"Error downloading dependencies: {str(e)}")
+#     except Exception as e:
+#         raise HTTPException(status_code=500, detail=str(e))
+#     finally:
+#         if os.path.exists(tmp_path):
+#             os.remove(tmp_path)
+#         if os.path.exists(log_file_path):
+#             os.remove(log_file_path)
+#         if os.path.exists(tar_path):
+#             os.remove(tar_path)        
+
 @app.post("/download-dependencies")
-async def download_dependencies(requirements_file: UploadFile = File(...)):
+async def download_dependencies(
+    requirements_file: UploadFile = File(...),
+    python_version: str = Form(None),
+):
+    tmp_path = None
+    log_file_path = None
+    tar_path = None
+    
     try:
+        # Use the current Python version if not specified
+        if not python_version:
+            python_version = "3.11"  # Adjust to your current version or use `sys.version_info`
+
+        python_bin, pip_bin = install_python_version(python_version)
+
+        # Validate the installed Python and pip
+        if not os.path.isfile(python_bin):
+            raise HTTPException(status_code=400, detail=f"Python {python_version} installation failed.")
+        
+        if not os.path.isfile(pip_bin):
+            raise HTTPException(status_code=400, detail=f"pip for Python {python_version} installation failed.")
+        
         with NamedTemporaryFile(delete=False) as tmp:
             tmp.write(requirements_file.file.read())
             tmp_path = tmp.name
@@ -39,11 +154,11 @@ async def download_dependencies(requirements_file: UploadFile = File(...)):
 
         log_file_path = "/tmp/dependencies/download.log"
         with open(log_file_path, "w") as log_file:
-            # Download dependencies
+            # Download dependencies using the specified Python version
             result = subprocess.run(
-                ["pip", "download", "-r", tmp_path, "-d", "/tmp/dependencies"],
-                # stdout=log_file,
-                # stderr=log_file
+                [python_bin, "-m", "pip", "download", "-r", tmp_path, "-d", "/tmp/dependencies"],
+                stdout=log_file,
+                stderr=log_file
             )
 
         if result.returncode != 0:
@@ -72,12 +187,13 @@ async def download_dependencies(requirements_file: UploadFile = File(...)):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
     finally:
-        if os.path.exists(tmp_path):
+        if tmp_path and os.path.exists(tmp_path):
             os.remove(tmp_path)
-        if os.path.exists(log_file_path):
+        if log_file_path and os.path.exists(log_file_path):
             os.remove(log_file_path)
-        if os.path.exists(tar_path):
-            os.remove(tar_path)        
+        if tar_path and os.path.exists(tar_path):
+            os.remove(tar_path)
+
 def download_docker_image(image_name, tag='latest', destination='/tmp/docker-images'):
     try:
         os.makedirs(destination, exist_ok=True)
@@ -142,6 +258,9 @@ async def read_root():
                 <label for="requirements_file">Requirements File:</label>
                 <input type="file" id="requirements_file" name="requirements_file" accept=".txt" required><br><br>
                 
+                <label for="python_version">Python Version</label>
+                <input type="text" id="python_version" name="python_version"><br><br>
+                
                 <input type="submit" id="pip-submit-button" value="Download Dependencies">
             </form>
             <div id="pip-message" style="display: none; margin-top: 10px;"></div>