Update app.py

app.py

@@ -55,6 +55,33 @@ def parse_changelog(text):
         components.append({'type': 'version', 'filename': parts[i].replace('##', '').strip(), 'content': parts[i+1].strip()})
     return components
 
+def parse_agent_action(text):
+    components = []
+    
+    # Extract Actions
+    action_pattern = re.compile(r'^### HF_ACTION: (.*)$', re.MULTILINE)
+    for match in action_pattern.finditer(text):
+        components.append({'type': 'action', 'filename': 'Agent Command', 'content': match.group(1).strip()})
+    
+    # Extract File Structure
+    structure_match = re.search(r'## File Structure\n(```[\s\S]*?```)', text)
+    if structure_match:
+        components.append({'type': 'structure', 'filename': 'Project Structure', 'content': structure_match.group(1).strip()})
+        
+    # Extract Files (using the existing logic from Repo2Markdown)
+    file_pattern = re.compile(r'### File: (.*?)\n([\s\S]*?)(?=\n### File:|\n## File Structure|\n### HF_ACTION:|\Z)', re.MULTILINE)
+    for match in file_pattern.finditer(text):
+        filename = match.group(1).strip()
+        content = match.group(2).strip()
+        is_code = "```" in content
+        components.append({'type': 'file', 'filename': filename, 'content': content, 'is_code_block': is_code})
+
+    # Instructions (Everything else)
+    if not components:
+        components.append({'type': 'instruction', 'filename': 'Instructions', 'content': text})
+        
+    return components
+
 @app.route('/parse', methods=['POST'])
 def parse_endpoint():
     text = request.form.get('markdown_text', '')
@@ -63,7 +90,9 @@ def parse_endpoint():
     if not text: return jsonify({'error': 'No text or file provided.'}), 400
     
     try:
-        if "## File Structure" in text and "### File:" in text:
+        if "### HF_ACTION:" in text or "File and Code Formatting:" in text:
+            format_name, components = "Agent Action", parse_agent_action(text)
+        elif "## File Structure" in text and "### File:" in text:
             format_name, components = "Repo2Markdown", parse_repo2markdown(text)
         elif re.search(r'^## \[\d+\.\d+\.\d+.*?\].*?$', text, flags=re.MULTILINE):
             format_name, components = "Changelog", parse_changelog(text)
@@ -71,6 +100,7 @@ def parse_endpoint():
             format_name, components = "Standard README", parse_standard_readme(text)
         else:
             format_name, components = "Unknown", [{'type': 'text', 'filename': 'Full Text', 'content': text}]
+            
         return jsonify({'format': format_name, 'components': components})
     except Exception as e:
         return jsonify({'error': f'Failed to parse: {e}'}), 500