Update classification.py

classification.py

@@ -7,6 +7,7 @@ import time
 import torch
 import concurrent.futures
 
+#testing 
 def hii():
     return "done"
     
@@ -48,34 +49,80 @@ def imageprocessing(image):
     return encoding
 
 # Run all models concurrently using threading
-def pipes(image):
+def pipes(image,categories):
     # Process the image once and reuse the encoding
     encoding = imageprocessing(image)
     
     # Access models from session state before threading
     top_wear_model = st.session_state.top_wear_model
+    full_wear_model=st.session_state.fullwear
+    bottom_wear_model=st.session_state.bottomwear_model
     pattern_model = st.session_state.pattern_model
     print_model = st.session_state.print_model
     sleeve_length_model = st.session_state.sleeve_length_model
+    #process ---------------------------------------------------------------------------upperwear--------------
+    if categories=="UpperBody":
+        # Define functions to run the models in parallel
+        with concurrent.futures.ThreadPoolExecutor() as executor:
+            futures = {
+                executor.submit(topwear, encoding, top_wear_model): "topwear",
+                executor.submit(patterns, encoding, pattern_model): "patterns",
+                executor.submit(prints, encoding, print_model): "prints",
+                executor.submit(sleevelengths, encoding, sleeve_length_model): "sleeve_length"
+            }
     
-    # Define functions to run the models in parallel
-    with concurrent.futures.ThreadPoolExecutor() as executor:
-        futures = {
-            executor.submit(topwear, encoding, top_wear_model): "topwear",
-            executor.submit(patterns, encoding, pattern_model): "patterns",
-            executor.submit(prints, encoding, print_model): "prints",
-            executor.submit(sleevelengths, encoding, sleeve_length_model): "sleeve_length"
-        }
-
-        results = {}
-        for future in concurrent.futures.as_completed(futures):
-            model_name = futures[future]
-            try:
-                results[model_name] = future.result()
-            except Exception as e:
-                st.error(f"Error in {model_name}: {str(e)}")
-                results[model_name] = None
+            results = {}
+            for future in concurrent.futures.as_completed(futures):
+                model_name = futures[future]
+                try:
+                    results[model_name] = future.result()
+                except Exception as e:
+                    st.error(f"Error in {model_name}: {str(e)}")
+                    results[model_name] = None
+    
+        return results
+    #process ---------------------------------------------------------------------------fullwear--------------
+    else if categories=="Wholebody":
+        # Define functions to run the models in parallel
+        with concurrent.futures.ThreadPoolExecutor() as executor:
+            futures = {
+                executor.submit(topwear, encoding, full_wear_model): "fullwear",
+                executor.submit(patterns, encoding, pattern_model): "patterns",
+                executor.submit(prints, encoding, print_model): "prints",
+                executor.submit(sleevelengths, encoding, sleeve_length_model): "sleeve_length"
+            }
+    
+            results = {}
+            for future in concurrent.futures.as_completed(futures):
+                model_name = futures[future]
+                try:
+                    results[model_name] = future.result()
+                except Exception as e:
+                    st.error(f"Error in {model_name}: {str(e)}")
+                    results[model_name] = None
+    
+        return results
+    #process ---------------------------------------------------------------------------bottomwear--------------
+    else if categories=="Lowerbody":
+        # Define functions to run the models in parallel
+        with concurrent.futures.ThreadPoolExecutor() as executor:
+            futures = {
+                executor.submit(topwear, encoding, full_wear_model): "fullwear",
+                executor.submit(patterns, encoding, pattern_model): "patterns",
+                executor.submit(prints, encoding, print_model): "prints",
+                executor.submit(sleevelengths, encoding, sleeve_length_model): "sleeve_length"
+            }
+    
+            results = {}
+            for future in concurrent.futures.as_completed(futures):
+                model_name = futures[future]
+                try:
+                    results[model_name] = future.result()
+                except Exception as e:
+                    st.error(f"Error in {model_name}: {str(e)}")
+                    results[model_name] = None
 
-    # Display the results
-    st.write(results)
-    return results
+        return results
+    else:
+        return {"invalid categorie":f"{categories} categorie not in process!"}
+