Update server.py

server.py

@@ -86,11 +86,16 @@ def extract_zip_if_needed(zip_path, dest_dir):
     print(f"✅ Extracted to: {dest_dir}")
 
 def print_directory_tree(path):
+    printstr = ''
     for root, dirs, files in os.walk(path):
         indent = '  ' * (root.count(os.sep) - path.count(os.sep))
-        print(f"{indent}📁 {os.path.basename(root)}/")
+        printstr += indent
+        printstr += os.path.basename(root)
+        printstr += '\n'
         for f in files:
-            print(f"{indent}  📄 {f}")
+            printstr += indent
+            printstr += f
+        return printstr
 
 
 def copy_zip_extract_and_report():
@@ -126,10 +131,11 @@ def copy_zip_extract_and_report():
             extract_zip_if_needed(dst_zip, extract_path)
 
     print("\n📦 Local hf_cache structure:")
-    print_directory_tree("./hf_cache")
+    printstr1 = print_directory_tree("./hf_cache")
 
     print("\n💾 Persistent /data/hf_cache structure:")
-    print_directory_tree("/data/hf_cache")
+    printstr2 = print_directory_tree("/data/hf_cache")
+    return printstr1 + '\n\n' + printstr2
 
 
 
@@ -138,11 +144,9 @@ def copy_and_extract():
     import io
     from contextlib import redirect_stdout
 
-    f = io.StringIO()
-    with redirect_stdout(f):
-        copy_zip_extract_and_report()
-    output = f.getvalue()
-    return {"message": "done", "log": output}
+    printstr = copy_zip_extract_and_report()
+    
+    return {"message": "done", "log": printstr}