Update app.py

app.py

@@ -126,3 +126,86 @@ async def upload_image(file: UploadFile = File(...)):
         return {"message": f"✅ {filename} uploaded successfully!"}
     except Exception as e:
         return {"error": f"❌ Upload failed: {str(e)}"}
+
+
+
+
+
+
+#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/"
+
+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:
+        all_files = list_repo_files(T_REPO_ID)
+        folder_prefix = T_FOLDER.rstrip("/") + "/"
+        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)
+            and "/" not in f[len(folder_prefix):]
+            and f.lower().endswith((".png", ".jpg", ".jpeg", ".webp"))
+        ]
+
+        if not image_files:
+            return {"error": f"No images found in folder '{T_FOLDER}'"}
+
+        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=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)
+
+        return {"message": f"Processed and uploaded {len(image_files)} images and text files.", "files": uploaded_files}
+
+    except Exception as e:
+        return {"error": str(e)}
+