Update main.py

main.py

@@ -1,68 +1,91 @@
 from flask import Flask, render_template, request, jsonify, send_file
-from flask_sqlalchemy import SQLAlchemy
-import pandas as pd
-from datetime import datetime
+import json
 import os
+from datetime import datetime
+import csv
+from io import StringIO
 
 app = Flask(__name__)
-app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///data/tickets.db'
-app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
-db = SQLAlchemy(app)
 
-class Ticket(db.Model):
-    id = db.Column(db.Integer, primary_key=True)
-    email = db.Column(db.String(120), nullable=False)
-    phone = db.Column(db.String(20), nullable=False)
-    name = db.Column(db.String(100), nullable=False)
-    ticket_count = db.Column(db.Integer, nullable=False)
-    ticket_number = db.Column(db.String(50), nullable=False)
-    country = db.Column(db.String(50), nullable=False)
-    region = db.Column(db.String(50), nullable=False)
-    timestamp = db.Column(db.DateTime, default=datetime.utcnow)
+# Ensure data directory exists
+os.makedirs('data', exist_ok=True)
 
-@app.route('/add_data/<email>/<phone>/<name>/<int:ticket_count>/<ticket_number>/<country>/<region>')
-def add_data(email, phone, name, ticket_count, ticket_number, country, region):
+# JSON file path
+DATA_FILE = 'data/tickets.json'
+
+def load_data():
     try:
-        new_ticket = Ticket(
-            email=email,
-            phone=phone,
-            name=name,
-            ticket_count=ticket_count,
-            ticket_number=ticket_number,
-            country=country,
-            region=region
-        )
-        db.session.add(new_ticket)
-        db.session.commit()
-        return jsonify({"status": "success", "message": "Ticket added successfully"})
-    except Exception as e:
-        return jsonify({"status": "error", "message": str(e)}), 400
+        with open(DATA_FILE, 'r') as f:
+            return json.load(f)
+    except (FileNotFoundError, json.JSONDecodeError):
+        return {"tickets": []}
+
+def save_data(data):
+    with open(DATA_FILE, 'w') as f:
+        json.dump(data, f, indent=2)
 
 @app.route('/')
-def dashboard():
-    tickets = Ticket.query.order_by(Ticket.timestamp.desc()).all()
-    return render_template('index.html', tickets=tickets)
+def index():
+    return render_template('index.html')
 
-@app.route('/download_excel')
-def download_excel():
-    try:
-        tickets = Ticket.query.all()
-        data = {
-            "Email": [t.email for t in tickets],
-            "Phone": [t.phone for t in tickets],
-            "Name": [t.name for t in tickets],
-            "Ticket Count": [t.ticket_count for t in tickets],
-            "Ticket Number": [t.ticket_number for t in tickets],
-            "Country": [t.country for t in tickets],
-            "Region": [t.region for t in tickets],
-            "Timestamp": [t.timestamp for t in tickets]
-        }
-        df = pd.DataFrame(data)
-        excel_path = os.path.join('data', 'tickets_export.xlsx')
-        df.to_excel(excel_path, index=False)
-        return send_file(excel_path, as_attachment=True)
-    except Exception as e:
-        return str(e), 500
+@app.route('/add_data/<email>/<phone>/<name>/<tickets_count>/<ticket_number>/<country>/<region>')
+def add_data(email, phone, name, tickets_count, ticket_number, country, region):
+    data = load_data()
+    
+    # Create new ticket entry
+    new_ticket = {
+        "email": email,
+        "phone": phone,
+        "name": name,
+        "tickets_count": int(tickets_count),
+        "ticket_number": ticket_number,
+        "country": country,
+        "region": region,
+        "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    }
+    
+    data["tickets"].append(new_ticket)
+    save_data(data)
+    
+    return jsonify({"status": "success", "message": "Ticket data added successfully"})
+
+@app.route('/get_tickets')
+def get_tickets():
+    data = load_data()
+    return jsonify(data)
+
+@app.route('/download_csv')
+def download_csv():
+    data = load_data()
+    
+    # Create CSV in memory
+    output = StringIO()
+    writer = csv.writer(output)
+    
+    # Write header
+    writer.writerow(["Email", "Phone", "Name", "Tickets Count", "Ticket Number", "Country", "Region", "Timestamp"])
+    
+    # Write data
+    for ticket in data["tickets"]:
+        writer.writerow([
+            ticket["email"],
+            ticket["phone"],
+            ticket["name"],
+            ticket["tickets_count"],
+            ticket["ticket_number"],
+            ticket["country"],
+            ticket["region"],
+            ticket["timestamp"]
+        ])
+    
+    output.seek(0)
+    
+    return send_file(
+        StringIO(output.getvalue()),
+        mimetype="text/csv",
+        as_attachment=True,
+        download_name="tickets_data.csv"
+    )
 
 if __name__ == '__main__':
     with app.app_context():