Update app.py

app.py

@@ -1,6 +1,6 @@
 # app.py
 from flask import Flask, request, render_template
-from parser import parse_python_file
+from parser import parse_python_code
 import os
 
 app = Flask(__name__)
@@ -8,23 +8,36 @@ app = Flask(__name__)
 @app.route('/', methods=['GET', 'POST'])
 def index():
     if request.method == 'POST':
-        if 'file' not in request.files:
-            return 'No file uploaded', 400
-
-        file = request.files['file']
-
-        if file.filename == '' or not file.filename.endswith('.py'):
-            return 'Invalid file type. Please upload a Python file.', 400
-
-        file_path = os.path.join('uploads', file.filename)
-        file.save(file_path)
-
-        parts = parse_python_file(file_path)
-        return render_template('results.html', parts=parts)
-
-    return render_template('upload.html')
+        parts = None
+        filename = 'unnamed.py'
+
+        # Handle file upload
+        if 'file' in request.files and request.files['file'].filename:
+            file = request.files['file']
+            if not file.filename.endswith('.py'):
+                return 'Invalid file type. Please upload a Python file.', 400
+            filename = file.filename
+            file_path = os.path.join('uploads', filename)
+            file.save(file_path)
+            with open(file_path, 'r') as f:
+                code = f.read()
+            parts = parse_python_code(code)
+
+        # Handle pasted code
+        elif 'code' in request.form and request.form['code'].strip():
+            code = request.form['code']
+            filename = request.form.get('filename', 'unnamed.py') or 'unnamed.py'
+            if not filename.endswith('.py'):
+                filename += '.py'
+            parts = parse_python_code(code)
+
+        if parts:
+            return render_template('results.html', parts=parts, filename=filename)
+        return 'No file or code provided', 400
+
+    return render_template('index.html')
 
 if __name__ == '__main__':
     if not os.path.exists('uploads'):
         os.makedirs('uploads')
-    app.run(host="0.0.0.0",port=7860)
+    app.run(port=7860)