Update app.py

app.py

@@ -1,106 +1,20 @@
-from flask import Flask, render_template, request, jsonify, send_file
-from .core import create_markdown_document
-import markdown
 import os
-import pkg_resources
 import sys
+import subprocess
 
-def find_template_path():
-    """Find the templates directory, either from installed package or source."""
-    try:
-        # Try to use pkg_resources for installed packages
-        template_path = pkg_resources.resource_filename("repo_to_md", "templates")
-    except Exception as e:
-        # Fallback for running from source (development)
-        current_dir = os.path.dirname(os.path.abspath(__file__))
-        template_path = os.path.join(current_dir, "../templates")
-        template_path = os.path.abspath(template_path)
+try:
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "git+https://github.com/broadfield-dev/repo_to_md.git#egg=repo_to_md[demo]"])
+except subprocess.CalledProcessError as e:
+    print(f"Failed to install package: {e}")
+    sys.exit(1)
 
-    # Check if the template path exists
-    if not os.path.exists(template_path):
-        # Try an alternative path within the package directory
-        package_dir = os.path.dirname(os.path.abspath(__file__))
-        alternative_path = os.path.join(package_dir, "templates")
-        if os.path.exists(alternative_path):
-            template_path = alternative_path
-        else:
-            # One last attempt: check if installed but in a different structure
-            site_packages = os.path.join(os.path.dirname(sys.executable), "site-packages")
-            installed_path = os.path.join(site_packages, "repo_to_md", "templates")
-            if os.path.exists(installed_path):
-                template_path = installed_path
-            else:
-                raise FileNotFoundError(
-                    f"Template directory not found at: {template_path}, {alternative_path}, or {installed_path}"
-                )
+from repo_to_md import create_markdown_document
 
-    return template_path
+try:
+    from repo_to_md import run_demo
+except ImportError as e:
+    print(f"Failed to import run_demo: {e}")
+    sys.exit(1)
 
-def run_demo(host="0.0.0.0", port=7860, debug=True):
-    # Get the correct template path
-    template_path = find_template_path()
-
-    # Create Flask app with the resolved template folder
-    app = Flask(__name__, template_folder=template_path)
-
-    @app.route('/')
-    def index():
-        return render_template('index.html')
-
-    @app.route('/process', methods=['POST'])
-    def process():
-        response_data = {'markdown': '', 'html': '', 'filename': '', 'error': None}
-        
-        try:
-            if 'files[]' in request.files:
-                files = request.files.getlist('files[]')
-                if not files:
-                    response_data['error'] = 'No files uploaded'
-                    return jsonify(response_data), 400
-                
-                markdown_content = create_markdown_document(files=files)
-                response_data['markdown'] = markdown_content
-                response_data['html'] = markdown.markdown(markdown_content)
-                response_data['filename'] = "uploaded_files_summary.md"
-            else:
-                repo_url = request.json.get('repo_url', '').strip()
-                if not repo_url:
-                    response_data['error'] = 'Please provide a repository URL or upload files'
-                    return jsonify(response_data), 400
-                
-                markdown_content = create_markdown_document(repo_url)
-                if markdown_content.startswith("Error:"):
-                    response_data['error'] = markdown_content
-                    return jsonify(response_data), 400
-                
-                response_data['markdown'] = markdown_content
-                response_data['html'] = markdown.markdown(markdown_content)
-                owner, repo = repo_url.rstrip('/').split('/')[-2:]
-                response_data['filename'] = f"{owner}_{repo}_summary.md"
-        
-        except Exception as e:
-            response_data['error'] = f"Server error processing request: {str(e)}"
-            return jsonify(response_data), 500
-        
-        return jsonify(response_data)
-
-    @app.route('/download', methods=['POST'])
-    def download():
-        markdown_content = request.json.get('markdown', '')
-        filename = request.json.get('filename', 'document.md')
-        
-        buffer = io.BytesIO()
-        buffer.write(markdown_content.encode('utf-8'))
-        buffer.seek(0)
-        
-        return send_file(
-            buffer,
-            as_attachment=True,
-            download_name=filename,
-            mimetype='text/markdown'
-        )
-
-    app.run(host=host, port=port, debug=debug)
-
-if __name__ == '__main__':
-    run_demo()
+# Run the demo (default: http://localhost:7860)
+run_demo()