Update app.py

app.py

@@ -1,21 +1,43 @@
 from flask import Flask, request, jsonify
 import cv2
 import sqlite3
+from datetime import datetime
 
 app = Flask(__name__)
 
+# Database setup
+conn = sqlite3.connect('materials.db', check_same_thread=False)
+c = conn.cursor()
+c.execute('''CREATE TABLE IF NOT EXISTS counts 
+             (id INTEGER PRIMARY KEY, name TEXT, count INTEGER, timestamp TEXT)''')
+conn.commit()
+
+# Dummy counting function (replace with actual camera logic)
+def detect_material():
+    # Yeh placeholder hai; actual OpenCV camera logic yahan aayega
+    return 5  # Simulated count
+
 @app.route('/count', methods=['POST'])
 def count_material():
-    name = request.json['name']
-    # Simulated count (replace with camera logic)
-    count = 5
-    conn = sqlite3.connect('materials.db')
-    c = conn.cursor()
+    data = request.json
+    name = data.get('name', 'Unknown')
+    
+    # Counting logic (camera se replace karo)
+    count = detect_material()
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    
+    # Save to database
     c.execute("INSERT INTO counts (name, count, timestamp) VALUES (?, ?, ?)", 
-              (name, count, "2025-02-28"))
+              (name, count, timestamp))
     conn.commit()
-    conn.close()
-    return jsonify({"material": name, "count": count})
+    
+    return jsonify({"material": name, "count": count, "timestamp": timestamp})
+
+@app.route('/view', methods=['GET'])
+def view_counts():
+    c.execute("SELECT * FROM counts")
+    results = c.fetchall()
+    return jsonify([{"id": r[0], "name": r[1], "count": r[2], "timestamp": r[3]} for r in results])
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)