use Firebase Firestore instead of SQLite

app.py

@@ -1,9 +1,7 @@
-from flask import Flask, render_template, request, jsonify, send_from_directory, make_response
-from flask_sqlalchemy import SQLAlchemy
-from werkzeug.utils import secure_filename
-from reportlab.lib.pagesizes import A4, landscape
-from reportlab.lib.units import cm, mm
-from reportlab.lib.colors import HexColor, white, black, Color
+from flask import Flask, render_template, request, jsonify, make_response
+from reportlab.lib.pagesizes import A4
+from reportlab.lib.units import cm
+from reportlab.lib.colors import HexColor
 from reportlab.pdfgen import canvas
 from reportlab.pdfbase import pdfmetrics
 from reportlab.pdfbase.ttfonts import TTFont
@@ -11,10 +9,18 @@ from PIL import Image as PILImage
 import arabic_reshaper
 from bidi.algorithm import get_display
 import io
-import os
 import base64
-import tempfile
-from datetime import datetime
+import requests
+import os
+
+FIREBASE_CONFIG = {
+    "apiKey": "AIzaSyAVacEYZNH0jIimwurEE0g4mTRUz1RvkT0",
+    "projectId": "waste-sorting-ai-36421",
+    "authDomain": "waste-sorting-ai-36421.firebaseapp.com",
+    "storageBucket": "waste-sorting-ai-36421.firebasestorage.app"
+}
+
+FIREBASE_URL = f"https://firestore.googleapis.com/v1/projects/{FIREBASE_CONFIG['projectId']}/databases/(default)/documents"
 
 pdfmetrics.registerFont(TTFont('Arabic', 'fonts/NotoNaskhArabic-Regular.ttf'))
 pdfmetrics.registerFont(TTFont('ArabicBold', 'fonts/NotoNaskhArabic-Bold.ttf'))
@@ -26,83 +32,93 @@ def reshape_arabic(text):
     return get_display(reshaped)
 
 app = Flask(__name__)
-app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///badges.db'
-app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
-app.config['UPLOAD_FOLDER'] = 'uploads'
-app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024
 
-os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
-
-db = SQLAlchemy(app)
-
-class Badge(db.Model):
-    id = db.Column(db.Integer, primary_key=True)
-    first_name = db.Column(db.String(100), nullable=False)
-    last_name = db.Column(db.String(100), nullable=False)
-    mission = db.Column(db.String(200))
-    photo_data = db.Column(db.Text)
-    bg_image_data = db.Column(db.Text)
-    background_color = db.Column(db.String(200), default='linear-gradient(135deg,#2d4a1e,#556b2f)')
-    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+def get_badges():
+    try:
+        resp = requests.get(f"{FIREBASE_URL}/badges", params={"orderBy": '"createdAt desc"'})
+        resp.raise_for_status()
+        docs = resp.json().get('documents', [])
+        badges = []
+        for doc in docs:
+            fields = doc.get('fields', {})
+            badges.append({
+                'id': doc['name'].split('/')[-1],
+                'firstName': fields.get('firstName', {}).get('stringValue', ''),
+                'lastName': fields.get('lastName', {}).get('stringValue', ''),
+                'mission': fields.get('mission', {}).get('stringValue', ''),
+                'photoUrl': fields.get('photoUrl', {}).get('stringValue', ''),
+                'bgImage': fields.get('bgImage', {}).get('stringValue', ''),
+                'backgroundColor': fields.get('backgroundColor', {}).get('stringValue', 'linear-gradient(135deg,#2d4a1e,#556b2f)'),
+                'createdAt': fields.get('createdAt', {}).get('integerValue', 0)
+            })
+        return badges
+    except Exception as e:
+        print(f"Error fetching badges: {e}")
+        return []
 
-    def to_dict(self):
-        return {
-            'id': self.id,
-            'firstName': self.first_name,
-            'lastName': self.last_name,
-            'name': f'{self.first_name} {self.last_name}',
-            'mission': self.mission,
-            'photoUrl': self.photo_data or 'https://via.placeholder.com/150',
-            'bgImage': self.bg_image_data,
-            'backgroundColor': self.background_color,
-            'createdAt': int(self.created_at.timestamp() * 1000) if self.created_at else 0
+def create_badge(data):
+    badge_data = {
+        'fields': {
+            'firstName': {'stringValue': data.get('firstName', '')},
+            'lastName': {'stringValue': data.get('lastName', '')},
+            'mission': {'stringValue': data.get('mission', '')},
+            'photoUrl': {'stringValue': data.get('photoUrl', '')},
+            'bgImage': {'stringValue': data.get('bgImage', '')},
+            'backgroundColor': {'stringValue': data.get('backgroundColor', 'linear-gradient(135deg,#2d4a1e,#556b2f)')},
+            'createdAt': {'integerValue': int(__import__('time').time() * 1000)}
         }
+    }
+    try:
+        resp = requests.post(f"{FIREBASE_URL}/badges", json=badge_data)
+        resp.raise_for_status()
+        return resp.json()
+    except Exception as e:
+        print(f"Error creating badge: {e}")
+        return None
+
+def delete_badge(badge_id):
+    try:
+        resp = requests.delete(f"{FIREBASE_URL}/badges/{badge_id}")
+        resp.raise_for_status()
+        return True
+    except Exception as e:
+        print(f"Error deleting badge: {e}")
+        return False
 
 @app.route('/')
 def index():
     return render_template('badge.html')
 
 @app.route('/api/badges', methods=['GET'])
-def get_badges():
-    badges = Badge.query.order_by(Badge.created_at.desc()).all()
-    return jsonify([b.to_dict() for b in badges])
+def get_badges_api():
+    badges = get_badges()
+    return jsonify(badges)
 
 @app.route('/api/badges', methods=['POST'])
-def create_badge():
+def create_badge_api():
     data = request.get_json()
-    
     if not data or not data.get('firstName') or not data.get('lastName'):
         return jsonify({'error': 'First name and last name are required'}), 400
     
-    badge = Badge(
-        first_name=data['firstName'],
-        last_name=data['lastName'],
-        mission=data.get('mission', ''),
-        photo_data=data.get('photoUrl'),
-        bg_image_data=data.get('bgImage'),
-        background_color=data.get('backgroundColor', 'linear-gradient(135deg,#2d4a1e,#556b2f)')
-    )
-    
-    db.session.add(badge)
-    db.session.commit()
-    
-    return jsonify(badge.to_dict()), 201
+    result = create_badge(data)
+    if result:
+        return jsonify({'message': 'Badge created', 'data': result}), 201
+    return jsonify({'error': 'Failed to create badge'}), 500
 
-@app.route('/api/badges/<int:badge_id>', methods=['DELETE'])
-def delete_badge(badge_id):
-    badge = Badge.query.get_or_404(badge_id)
-    db.session.delete(badge)
-    db.session.commit()
-    return jsonify({'message': 'Badge deleted'})
+@app.route('/api/badges/<badge_id>', methods=['DELETE'])
+def delete_badge_api(badge_id):
+    if delete_badge(badge_id):
+        return jsonify({'message': 'Badge deleted'})
+    return jsonify({'error': 'Failed to delete badge'}), 500
 
 @app.route('/api/export/print')
 def export_print():
-    badges = Badge.query.order_by(Badge.created_at.desc()).all()
+    badges = get_badges()
     
     badges_html = ""
     for badge in badges:
-        bg_style = f"background-image: url({badge.bg_image_data}); background-size: cover; background-position: center;" if badge.bg_image_data else f"background: {badge.background_color};"
-        photo = badge.photo_data if badge.photo_data else 'https://via.placeholder.com/150'
+        bg_style = f"background-image: url({badge.get('bgImage')}); background-size: cover; background-position: center;" if badge.get('bgImage') else f"background: {badge.get('backgroundColor', '#2d4a1e')};"
+        photo = badge.get('photoUrl', 'https://via.placeholder.com/150')
         
         badges_html += f"""
         <div class="badge-item">
@@ -112,11 +128,11 @@ def export_print():
                 <div class="badge-content">
                     <div class="badge-scout-icon">⚜️</div>
                     <img class="badge-photo" src="{photo}" alt="">
-                    <div class="badge-name">{badge.first_name} {badge.last_name}</div>
+                    <div class="badge-name">{badge.get('firstName', '')} {badge.get('lastName', '')}</div>
                 </div>
                 <div class="badge-mission-wrap">
                     <div class="badge-mission-label">المهمة</div>
-                    <div class="badge-mission-value">{badge.mission or '—'}</div>
+                    <div class="badge-mission-value">{badge.get('mission', '—')}</div>
                 </div>
             </div>
         </div>
@@ -171,7 +187,7 @@ body {{ font-family: 'Tajawal', sans-serif; background: white; padding: 20px; }}
 
 @app.route('/api/export/pdf')
 def export_pdf():
-    badges = Badge.query.order_by(Badge.created_at.desc()).all()
+    badges = get_badges()
     
     badge_width = 8.5 * cm
     badge_height = 12 * cm
@@ -217,7 +233,7 @@ def export_pdf():
         c.setLineWidth(3)
         c.roundRect(2, 2, badge_width - 4, badge_height - 4, 6, fill=0, stroke=1)
         
-        bg_buf = get_image_buffer(badge.bg_image_data, size=(int(badge_width/cm*100), int(badge_height/cm*100)))
+        bg_buf = get_image_buffer(badge.get('bgImage', ''), size=(int(badge_width/cm*100), int(badge_height/cm*100)))
         if bg_buf:
             c.drawImage(bg_buf, 0, 0, width=badge_width, height=badge_height, preserveAspectRatio=False, mask='auto')
             c.setFillColor(HexColor('#000000'))
@@ -241,7 +257,7 @@ def export_pdf():
         c.drawImage(logo_buf, badge_width/2 - 0.7*cm, center_y + 1.5*cm, width=1.4*cm, height=1.4*cm)
         
         photo_size = 2.2 * cm
-        photo_buf = get_image_buffer(badge.photo_data, size=(200, 200))
+        photo_buf = get_image_buffer(badge.get('photoUrl', ''), size=(200, 200))
         if photo_buf:
             c.saveState()
             c.beginPath()
@@ -259,7 +275,7 @@ def export_pdf():
             c.setFont('ArabicBold', 24)
             c.drawCentredString(badge_width/2, center_y - 0.3*cm, "📷")
         
-        name = f"{badge.first_name} {badge.last_name}"
+        name = f"{badge.get('firstName', '')} {badge.get('lastName', '')}"
         c.setFillColor(HexColor('#ffffff'))
         c.setFont('ArabicBold', 12)
         c.setFillColor(HexColor('#000000'))
@@ -272,7 +288,9 @@ def export_pdf():
         c.setFillColor(HexColor('#3d3010'))
         c.rect(0, 0, badge_width, 1.3*cm, fill=1, stroke=0)
         draw_text_rtl(c, "المهمة", badge_width/2, 0.9*cm, 'Arabic', 9, HexColor('#c8a96e'))
-        mission_text = badge.mission[:20] if badge.mission and len(badge.mission) > 20 else (badge.mission or "—")
+        mission_text = badge.get('mission', '—')
+        if len(mission_text) > 20:
+            mission_text = mission_text[:20]
         draw_text_rtl(c, mission_text, badge_width/2, 0.35*cm, 'ArabicBold', 10, HexColor('#f5e6b0'))
         
         c.restoreState()
@@ -301,10 +319,5 @@ def export_pdf():
     
     return response
 
-def init_db():
-    with app.app_context():
-        db.create_all()
-
 if __name__ == '__main__':
-    init_db()
     app.run(debug=True, host='0.0.0.0', port=5000)

requirements.txt

@@ -1,7 +1,7 @@
 flask>=2.3.0
-flask-sqlalchemy>=3.0.0
 werkzeug>=2.3.0
 reportlab>=4.0.0
 Pillow>=10.0.0
 arabic-reshaper>=3.0.0
 python-bidi>=0.6.0
+requests>=2.31.0