Update app_logic.py

app_logic.py

@@ -62,8 +62,6 @@ def load_token_from_image_and_set_env(image_pil_object: Image.Image, password: s
 
 
 
-import re
-
 def process_commented_markdown(commented_input):
     """Process a commented markdown string by stripping '# ' from lines where present if '# # Space:' is detected."""
     # Check for '# # Space:' or '# Space:' to identify a space definition
@@ -78,7 +76,6 @@ def process_commented_markdown(commented_input):
         # Detect code block boundaries (handle '# ```' or '```')
         if line.strip() in ("# ```", "```"):
             in_code_block = not in_code_block
-            # Preserve backticks as-is (strip '# ' if present)
             cleaned_line = line.lstrip("# ").rstrip()
             cleaned_lines.append(cleaned_line)
             continue
@@ -94,36 +91,34 @@ def process_commented_markdown(commented_input):
 
     return "\n".join(cleaned_lines)
 
+
+
 def parse_markdown(markdown_input):
     """Parse markdown input to extract space details and file structure."""
-    # Preprocess to remove comments
+    # Preprocess to handle commented markdown
     cleaned_markdown = process_commented_markdown(markdown_input)
-    space_info = {"repo_name": "", "owner": "", "files": []}
+    space_info = {"repo_name": "", "owner": "", "sdk": "gradio", "files": []}
     current_file = None
     file_content = []
     in_file_content = False
     in_code_block = False
-    language = None
 
     lines = cleaned_markdown.strip().split("\n")
-    i = 0
-    while i < len(lines):
-        line = lines[i].strip()
+    for line in lines:
+        line = line.strip()
 
         # Extract space name
         if line.startswith("# Space:"):
             space_info["repo_name"] = line.replace("# Space:", "").strip()
             if "/" in space_info["repo_name"]:
                 space_info["owner"], space_info["repo_name"] = space_info["repo_name"].split("/", 1)
-            i += 1
             continue
 
         # Skip file structure header
         if line.startswith("## File Structure"):
-            i += 1
             continue
 
-        # Parse file structure
+        # Detect file in structure
         if line.startswith("📄") or line.startswith("📁"):
             if current_file and file_content and in_file_content:
                 space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})
@@ -131,11 +126,9 @@ def parse_markdown(markdown_input):
             current_file = line[2:].strip()
             in_file_content = False
             in_code_block = False
-            language = None
-            i += 1
             continue
 
-        # Start of a new file content section
+        # Detect file content section
         if line.startswith("### File:"):
             if current_file and file_content and in_file_content:
                 space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})
@@ -143,42 +136,40 @@ def parse_markdown(markdown_input):
             current_file = line.replace("### File:", "").strip()
             in_file_content = True
             in_code_block = False
-            language = None
-            i += 1
-
-            # Handle content (code block or plain text)
-            if i < len(lines) and lines[i].strip() == "```":
-                in_code_block = True
-                i += 1
-                # Check for language identifier
-                if i < len(lines) and lines[i].strip() and not lines[i].strip().startswith("```"):
-                    language = lines[i].strip()
-                    i += 1
-                # Collect content until closing backtick
-                while i < len(lines) and lines[i].strip() != "```":
-                    file_content.append(lines[i].rstrip())
-                    i += 1
-                if i < len(lines) and lines[i].strip() == "```":
-                    i += 1
-                in_code_block = False
-                in_file_content = False
+            continue
+
+        # Handle code block boundaries
+        if in_file_content and line == "```":
+            in_code_block = not in_code_block
+            if in_code_block:
+                file_content = []  # Start fresh for code block content
+            else:
+                # End of code block, save content if any
+                if current_file and file_content:
+                    space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})
+                    file_content = []
+                    in_file_content = False
+            continue
+
+        # Collect content
+        if in_file_content:
+            if in_code_block:
+                # Skip language identifier (first non-empty line after opening backtick)
+                if not file_content and line and not line.startswith("```"):
+                    continue  # Skip language line (e.g., `md`, `py`)
+                file_content.append(line)
             else:
                 # Handle non-code-block content (e.g., binary file)
-                while i < len(lines) and not (
-                    lines[i].strip().startswith("### File:")
-                    or lines[i].strip().startswith("#")
-                    or lines[i].strip().startswith("##")
-                    or lines[i].strip() == "```"
-                ):
-                    if lines[i].strip():
-                        file_content.append(lines[i].rstrip())
-                    i += 1
-                in_file_content = False
-
-        else:
-            i += 1
-
-    # Save the last file if present
+                if line:
+                    file_content.append(line)
+                else:
+                    # End non-code-block content at empty line or next section
+                    if current_file and file_content:
+                        space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})
+                        file_content = []
+                        in_file_content = False
+
+    # Append the last file
     if current_file and file_content and in_file_content:
         space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})