New format for image storage

app.py

@@ -11,12 +11,17 @@ import urllib.request
 from PIL import Image
 import requests
 import matplotlib.pyplot as plt
+from pathlib import Path
+from datetime import datetime
 
 from transformers import CLIPProcessor, CLIPModel
 
 model = CLIPModel.from_pretrained("patrickjohncyh/fashion-clip")
 processor = CLIPProcessor.from_pretrained("patrickjohncyh/fashion-clip")
 
+static_dir = Path('./static')
+static_dir.mkdir(parents=True, exist_ok=True)
+
 # sys.path.insert(1, 'C:/Users/Alexandre/Documents/University/5_Ano/Estagio/repos_1')
 
 # Create custom Color objects for our primary, secondary, and neutral colors
@@ -215,6 +220,18 @@ with gr.Blocks(theme=theme, title="New Look", css=css) as demo:
             gr.Markdown()
             gr.Markdown()
 
+    def load(image_input):
+        file_name = f"{datetime.utcnow().strftime('%s')}.jpg"
+        file_path = static_dir / file_name
+        print(file_path)
+        image_input.save("img_path.jpg")
+        with open(file_path, "w") as f:
+            f.write("""
+                    {% load static %}
+                    <img src="{% static 'img_path.jpg' %}" alt="My image">
+            """)
+        link = f'<a href="file={file_path}" target="_blank">{file_name}</a>'
+        return link
 
     def search(text_input, image_input, image_path, text_relevance, image_relevance, best_seller_score_weight):
         if text_input == "" and image_input == None:
@@ -251,7 +268,12 @@ with gr.Blocks(theme=theme, title="New Look", css=css) as demo:
         return bar_plot, labels_probs
         
 
+    # search_button.click(
+    #     search, [text_input, image_input, image_path, text_relevance, image_relevance, best_seller_score_weight], [image_res_1, image_res_2, image_res_3, image_res_4, image_res_5, response]
+    # )
     search_button.click(
+        load, image_input, image_path
+    ).then(
         search, [text_input, image_input, image_path, text_relevance, image_relevance, best_seller_score_weight], [image_res_1, image_res_2, image_res_3, image_res_4, image_res_5, response]
     )