Update app_logic.py

app_logic.py

@@ -1,13 +1,12 @@
 import os
-import re
 import tempfile
-import shutil
+import gradio as gr
 from huggingface_hub import create_repo, upload_folder, whoami
 import logging
 from pathlib import Path
 from PIL import Image
-
 from keylock import core as keylock_core
+from repo_to_md.core import markdown_to_files
 
 logging.basicConfig(
     level=logging.INFO, 
@@ -15,27 +14,20 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
-# --- NEW FUNCTION TO GET USERNAME ---
 def get_username_from_token(api_token: str):
-    """
-    Fetches the username associated with a Hugging Face API token.
-    Returns the username on success, or an empty string on failure.
-    """
     if not api_token:
         logger.info("API token is empty, cannot fetch username.")
         return ""
-
     try:
         logger.info("Attempting to fetch username from provided API token.")
         user_info = whoami(token=api_token)
         username = user_info.get('name')
-        
         if username:
             logger.info(f"Successfully fetched username: {username}")
             return username
         else:
             logger.warning("Token was valid, but no username found in whoami() response.")
-            return "" # Return empty if name key doesn't exist
+            return ""
     except Exception as e:
         logger.error(f"Failed to validate token and fetch username: {e}")
         return ""
@@ -57,12 +49,12 @@ def extract_markdown_from_image(repo_image: Image.Image, image_password: str):
         if not markdown_content:
             return None, "Error: Extraction succeeded, but no markdown content was found in the image."
         logger.info("Successfully extracted markdown content.")
-        return markdown_content, None
+        return markdown_content, result.get('status', "Extraction successful.")
     except Exception as e:
         logger.error(f"Failed to extract from image: {e}")
         return None, f"Error during extraction: {e}"
 
-def parse_markdown(markdown_input):
+def parse_markdown_for_build(markdown_input):
     space_info = {"files": []}
     current_file_path = None; current_file_content_lines = []
     in_file_definition = False; in_code_block = False
@@ -106,7 +98,7 @@ def create_space(ui_api_token_from_textbox, space_name_ui, owner_ui, sdk_ui, mar
     try:
         resolved_api_token, token_err = _get_api_token(ui_api_token_from_textbox)
         if token_err: return token_err
-        space_info = parse_markdown(markdown_input)
+        space_info = parse_markdown_for_build(markdown_input)
         if not space_info["files"]: return "Error: No files were found after parsing the markdown from the image."
         with tempfile.TemporaryDirectory() as temp_dir:
             repo_staging_path = Path(temp_dir) / "repo_staging"
@@ -131,8 +123,47 @@ def create_space(ui_api_token_from_textbox, space_name_ui, owner_ui, sdk_ui, mar
         return f"Error during Space creation: {str(e)}"
 
 def build_space_from_image(api_token, repo_image, image_password, new_space_name, owner, sdk):
-    markdown_string, error = extract_markdown_from_image(repo_image, image_password)
-    if error:
-        return error
+    markdown_string, status_msg = extract_markdown_from_image(repo_image, image_password)
+    if not markdown_string:
+        return status_msg
     result = create_space(api_token, new_space_name, owner, sdk, markdown_string)
-    return result
+    return result
+
+def preview_image_contents(repo_image, image_password):
+    markdown_string, status_msg = extract_markdown_from_image(repo_image, image_password)
+    if not markdown_string:
+        return status_msg, gr.update(visible=False), gr.update(choices=[], value=None), "", []
+    try:
+        file_data, _ = markdown_to_files(markdown_string)
+        if isinstance(file_data, str) or not file_data:
+            raise ValueError("Markdown parsing failed or returned no files.")
+        
+        filenames = [f['filename'] for f in file_data]
+        if not filenames:
+            status = "✅ Preview generated, but no files were found in the markdown structure."
+            return status, gr.update(visible=False), gr.update(choices=[], value=None), "", []
+
+        first_filename = filenames
+        first_content = file_data[0]['content']
+        
+        status = f"✅ Preview generated successfully. Found {len(filenames)} file(s)."
+        
+        return (
+            status,
+            gr.update(visible=True),
+            gr.update(choices=filenames, value=first_filename),
+            first_content,
+            file_data
+        )
+    except Exception as e:
+        logger.error(f"Error parsing markdown for preview: {e}", exc_info=True)
+        error_msg = f"❌ Error: Could not parse files from the image's content. {e}"
+        return error_msg, gr.update(visible=False), gr.update(choices=[], value=None), "", []
+
+def update_file_preview(selected_filename, all_file_data):
+    if not selected_filename or not all_file_data:
+        return ""
+    selected_file = next((f for f in all_file_data if f['filename'] == selected_filename), None)
+    if selected_file:
+        return selected_file.get('content', 'File content not found.')
+    return f"Select file to view contents."