up

app.py

@@ -2,8 +2,10 @@ import gradio as gr
 from convert_diffusion_to_gguf import SUPPORTED_ARCHS, qconfig_map, convert
 from huggingface_hub import create_repo, upload_file
 from argparse import Namespace
+from io import StringIO
 from pathlib import Path
 
+log_stream = StringIO()
 
 def upload(args):
     url = ""
@@ -17,6 +19,8 @@ def upload(args):
 
 
 def go_gguf(model_repo_id, subfolder, arch, outtype, outfile_name, bigendian, verbose, host_repo_id, hf_token, progress=gr.Progress(track_tqdm=True)):
+    log_stream.truncate(0)
+    log_stream.seek(0)
     args = Namespace(
         model=Path(model_repo_id),
         subfolder=subfolder,
@@ -35,11 +39,11 @@ def go_gguf(model_repo_id, subfolder, arch, outtype, outfile_name, bigendian, ve
         progress(0.8, desc="✅ Conversion Complete. Starting upload...")
         url = upload(args)
         if url:
-            return f"### ✅ Success!\n\nUploaded to: [{url}]({url})"
+            return log_stream.getvalue(), f"### ✅ Success!\n\nUploaded to: [{url}]({url})"
         else:
-            return "### ✅ Conversion Complete!\n\n(File was not uploaded as no repo/token was provided)"
+            return log_stream.getvalue(), "### ✅ Conversion Complete!\n\n(File was not uploaded as no repo/token was provided)"
     except Exception as e:
-        return str(e)
+        return log_stream.getvalue(), str(e)
 
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -70,6 +74,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             convert_btn = gr.Button("Convert & Upload", variant="primary")
 
         with gr.Column(scale=2):
+            gr.Markdown("### 📝 Logs")
+            logs_output = gr.Textbox(label="Conversion Logs", lines=25, max_lines=25, interactive=False, autoscroll=True)
             gr.Markdown("### 🚀 Result")
             url_output = gr.Markdown()
 
@@ -104,7 +110,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     convert_btn.click(
         fn=go_gguf,
         inputs=[model_repo_id, subfolder, arch, outtype, outfile_name, bigendian, verbose, host_repo_id, hf_token],
-        outputs=[url_output],
+        outputs=[logs_output, url_output],
     )
 
 if __name__ == "__main__":