Update app.py

app.py

@@ -134,14 +134,18 @@ async def upload_image(file: UploadFile = File(...)):
 
 
 #Tranining Data set start fitering data for traninig
+
+
+
 T_REPO_ID = "rahul7star/ohamlab"
 T_FOLDER = "demo"
-T_BASE_URL = f"https://huggingface.co/{REPO_ID}/resolve/main/"
+T_BASE_URL = f"https://huggingface.co/{T_REPO_ID}/resolve/main/"
 
 DESCRIPTION_TEXT = (
     "Ra3hul is wearing a black jacket over a striped white t-shirt with blue jeans. "
     "He is standing near a lake with his arms spread wide open, with mountains and cloudy skies in the background."
 )
+
 @app.post("/filter-images")
 def filter_and_rename_images():
     try:
@@ -150,7 +154,6 @@ def filter_and_rename_images():
         filter_folder = f"filter-{T_FOLDER}"
         filter_prefix = filter_folder + "/"
 
-        # Get images only directly in original folder
         image_files = [
             f for f in all_files
             if f.startswith(folder_prefix)
@@ -163,50 +166,40 @@ def filter_and_rename_images():
 
         uploaded_files = []
 
-        with tempfile.TemporaryDirectory() as tmpdir:
-            for idx, orig_path in enumerate(image_files, start=1):
-                # Download original image to temp dir
-                local_path = hf_hub_download(repo_id=REPO_ID, filename=orig_path)
-
-                # Determine new image filename - all renamed to .jpeg (can keep original ext if you want)
-                new_image_name = f"image{idx}.jpeg"
-                new_image_path = os.path.join(tmpdir, new_image_name)
-
-                # Convert/Copy image to new path (simple copy for now)
-                # If needed, use PIL to convert formats. Here, just copy
-                with open(local_path, "rb") as src_file, open(new_image_path, "wb") as dst_file:
-                    dst_file.write(src_file.read())
-
-                # Upload renamed image to HF under filter folder
-                upload_file(
-                    path_or_fileobj=new_image_path,
-                    path_in_repo=filter_prefix + new_image_name,
-                    repo_id=T_REPO_ID,
-                    repo_type="model",
-                    commit_message=f"Upload renamed image {new_image_name} to {filter_folder}",
-                    token=True,
-                )
-                uploaded_files.append(filter_prefix + new_image_name)
-
-                # Create corresponding .txt file with description
-                txt_filename = f"image{idx}.txt"
-                txt_path = os.path.join(tmpdir, txt_filename)
-                with open(txt_path, "w") as txt_file:
-                    txt_file.write(DESCRIPTION_TEXT)
-
-                # Upload text file
-                upload_file(
-                    path_or_fileobj=txt_path,
-                    path_in_repo=filter_prefix + txt_filename,
-                    repo_id=T_REPO_ID,
-                    repo_type="model",
-                    commit_message=f"Upload text file {txt_filename} to {filter_folder}",
-                    token=True,
-                )
-                uploaded_files.append(filter_prefix + txt_filename)
+        for idx, orig_path in enumerate(image_files, start=1):
+            # Download file content into bytes (no local disk)
+            local_path = hf_hub_download(repo_id=T_REPO_ID, filename=orig_path)
+            with open(local_path, "rb") as f:
+                file_bytes = f.read()
+
+            new_image_name = f"image{idx}.jpeg"  # rename all to .jpeg
+
+            # Upload image bytes directly from memory
+            upload_file(
+                path_or_fileobj=io.BytesIO(file_bytes),
+                path_in_repo=filter_prefix + new_image_name,
+                repo_id=T_REPO_ID,
+                repo_type="model",
+                commit_message=f"Upload renamed image {new_image_name} to {filter_folder}",
+                token=True,
+            )
+            uploaded_files.append(filter_prefix + new_image_name)
+
+            # Create and upload text file from string bytes
+            txt_filename = f"image{idx}.txt"
+            upload_file(
+                path_or_fileobj=io.BytesIO(DESCRIPTION_TEXT.encode("utf-8")),
+                path_in_repo=filter_prefix + txt_filename,
+                repo_id=T_REPO_ID,
+                repo_type="model",
+                commit_message=f"Upload text file {txt_filename} to {filter_folder}",
+                token=True,
+            )
+            uploaded_files.append(filter_prefix + txt_filename)
 
         return {"message": f"Processed and uploaded {len(image_files)} images and text files.", "files": uploaded_files}
 
     except Exception as e:
         return {"error": str(e)}
+