Update app.py

app.py

@@ -47,13 +47,16 @@ def preprocess_and_create_stylecloud(file_path, output_name='stylecloud.png',
     plt.show()
 
 def create_stylecloud(text, language, icon):
-    output_file = "stylecloud.png"
+        output_file = "static/stylecloud.png"
     
     stylecloud.gen_stylecloud(text=text,
                               icon_name=icon,
                               output_name=output_file)
-    
-    return output_file
+     # Dosyayı Base64 olarak kodla
+    with open(output_file, "rb") as file:
+        data = file.read()
+    encoded = base64.b64encode(data).decode()
+    return encoded
 
 st.title("WordCloud Creator")
 
@@ -92,10 +95,17 @@ if file is not None:
     icon = [icon for icon, label in icon_options if label == icon_selection][0]
     
     if st.button("Create"):
-        output_file = create_stylecloud(text, language, icon)
-        st.markdown(f"### [Download WordCloud](/static/{output_file})")
+        encoded_image = create_stylecloud(text, language, icon)
+        st.markdown(
+        f"""
+        <a href="data:application/octet-stream;base64,{encoded_image}" download="stylecloud.png">
+            <button>Download WordCloud</button>
+        </a>
+        """,
+        unsafe_allow_html=True
+    )
 
-        image = Image.open(output_file)
+        image = Image.open("static/stylecloud.png")
         st.image(image, caption='WordCloud', use_column_width=True)
         # Ensure the file is deleted after display
         if os.path.exists(output_file):