Update app.py

app.py

@@ -196,14 +196,11 @@ def process_video_and_save(uploaded_file):
                             license_plate_text = model_ocr.chat(processor, temp_image_path, ocr_type='ocr')
                             filtered_text = filter_license_plate_text(license_plate_text)
                             # Check if the license plate is already detected and saved
-                            if filtered_text:
-                                # Get the email from the database
-                                email = get_vehicle_information(filtered_text, lane_violation, helmet_violation, violation_image_path, "Riyadh")
-                            
+                            if filtered_text:                            
                                 # Add the license plate and its violations to the violations dictionary
                                 if filtered_text not in violations_dict:
                                     violations_dict[filtered_text] = violation_type  #{"1234AB":[no_Helmet,In_red_Lane]}
-                                    send_email(filtered_text, violation_image_path, ', '.join(violation_type), email)
+                                    send_email(filtered_text, violation_image_path, ', '.join(violation_type))
                                 else:
                                     # Update the violations for the license plate if new ones are found
                                     current_violations = set(violations_dict[filtered_text])  # no helmet
@@ -213,7 +210,7 @@ def process_video_and_save(uploaded_file):
                                     # If new violations are found, update and send email
                                     if updated_violations != violations_dict[filtered_text]:
                                         violations_dict[filtered_text] = updated_violations
-                                        send_email(filtered_text, violation_image_path, ', '.join(updated_violations), email)
+                                        send_email(filtered_text, violation_image_path, ', '.join(updated_violations))
 
                             # Draw OCR text (English and Arabic) on the original frame
                             arabic_text = convert_to_arabic(filtered_text)
@@ -320,177 +317,6 @@ def draw_text_pil(img, text, position, font_path, font_size, color):
     img_np = cv2.cvtColor(np.array(img_pil), cv2.COLOR_RGB2BGR)
     return img_np
 
-import sqlite3
-from datetime import datetime
-from sqlalchemy import create_engine
-from sqlalchemy.orm import sessionmaker
-from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy import Column, String, Integer, Boolean
-
-def get_vehicle_information(detected_license_plate, lane_violation, no_helmet, image_link, city):
-    # Get current date and time
-    current_datetime = datetime.now()
-    current_date = current_datetime.strftime('%Y-%m-%d')
-    current_time = current_datetime.strftime('%H:%M:%S')
-
-    # Connect to SQLite database (motorbike_detections.db)
-    try:
-        motorbike_conn = sqlite3.connect('motorbike_detections.db')
-        motorbike_cursor = motorbike_conn.cursor()
-
-        # Check if the detection already exists
-        check_query = '''
-            SELECT DetectionID FROM MotorbikeDetections
-            WHERE LicensePlate = ? AND Date = ? AND Time = ? AND ImageLink = ?
-        '''
-        motorbike_cursor.execute(check_query, (detected_license_plate, current_date, current_time, image_link))
-        existing_detection = motorbike_cursor.fetchone()
-
-        if existing_detection:
-            print(f"Detection for license plate {detected_license_plate} at {current_date} {current_time} already exists.")
-        else:
-            # Insert the new detection record
-            insert_query = '''
-                INSERT INTO MotorbikeDetections (Date, Time, City, LicensePlate, LaneViolation, NoHelmet, ImageLink)
-                VALUES (?, ?, ?, ?, ?, ?, ?)
-            '''
-            motorbike_cursor.execute(insert_query, (
-                current_date,
-                current_time,
-                city,
-                detected_license_plate,
-                int(lane_violation),  # Convert boolean to integer (1 or 0)
-                int(no_helmet),        # Convert boolean to integer (1 or 0)
-                image_link
-            ))
-
-            # Commit the transaction
-            motorbike_conn.commit()
-            print(f"Detection data for license plate {detected_license_plate} inserted successfully.")
-
-    except sqlite3.IntegrityError as e:
-        print(f"Integrity Error: {e}. This detection may already exist.")
-    except sqlite3.Error as e:
-        print(f"An error occurred while inserting detection data: {e}")
-        motorbike_conn.rollback()
-    finally:
-        # Close the motorbike detections database connection
-        motorbike_conn.close()
-
-    # Retrieve email from 'vehicle_information.db'
-    try:
-        # Create an engine and session for SQLAlchemy
-        engine = create_engine('sqlite:///vehicle_information.db')
-        Session = sessionmaker(bind=engine)
-        session = Session()
-
-        # Query the VehicleInformation table for the detected license plate
-        vehicle_info = session.query(VehicleInformation).filter_by(license_plate=detected_license_plate).first()
-
-        if vehicle_info:
-            print(f"Email found for license plate {detected_license_plate}: {vehicle_info.email}")
-            return vehicle_info.email
-        else:
-            print(f"No vehicle information found for license plate {detected_license_plate}.")
-            return None
-
-    except Exception as e:
-        print(f"An error occurred while retrieving vehicle information: {e}")
-        return None
-
-    finally:
-        # Close the SQLAlchemy session
-        session.close()
-
-import sqlite3
-from datetime import datetime
-
-def setup_motorbike_detections_db():
-    # Connect to SQLite database (or create it if it doesn't exist)
-    conn = sqlite3.connect('motorbike_detections.db')
-    cursor = conn.cursor()
-
-    # Drop the old table if it exists, useful for restructuring
-    cursor.execute('DROP TABLE IF EXISTS MotorbikeDetections')
-
-    # Create the new table with a unique constraint on LicensePlate, Date, Time, and ImageLink
-    cursor.execute('''
-    CREATE TABLE IF NOT EXISTS MotorbikeDetections (
-        DetectionID INTEGER PRIMARY KEY AUTOINCREMENT,
-        Date DATE NOT NULL,
-        Time TIME NOT NULL,
-        City VARCHAR(100),
-        LicensePlate VARCHAR(100),
-        LaneViolation BOOLEAN NOT NULL,
-        NoHelmet BOOLEAN NOT NULL,
-        ImageLink VARCHAR(255),
-        UNIQUE(LicensePlate, Date, Time, ImageLink)
-    )
-    ''')
-
-    # Commit changes
-    conn.commit()
-
-    # Close the connection
-    conn.close()
-
-
-from sqlalchemy import create_engine, Column, String, Integer
-from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import sessionmaker
-
-# Define the base for model creation
-Base = declarative_base()
-
-# Define the VehicleInformation table using SQLAlchemy ORM
-class VehicleInformation(Base):
-    __tablename__ = 'VehicleInformation'
-
-    id = Column(Integer, primary_key=True, autoincrement=True)
-    license_plate = Column(String(100), unique=True, nullable=False)
-    email = Column(String(255), nullable=False)
-    phone_number = Column(String(15), nullable=False)
-    driver_id = Column(String(50), nullable=False)  # Government ID or identity number
-
-def setup_database():
-    # Create an SQLite database (this could be any database like PostgreSQL, MySQL, etc.)
-    engine = create_engine('sqlite:///vehicle_information.db')
-
-    # Drop existing tables and recreate them (for ensuring clean database on rerun)
-    Base.metadata.drop_all(engine)
-    Base.metadata.create_all(engine)
-
-    # Create a session to interact with the database
-    Session = sessionmaker(bind=engine)
-    session = Session()
-
-    # Insert some dummy data into the VehicleInformation table
-    vehicle_data_1 = VehicleInformation(
-        license_plate='1234 AB',
-        email='[email protected]',
-        phone_number='0559947203',
-        driver_id='ID1110000000'
-    )
-
-    vehicle_data_2 = VehicleInformation(
-        license_plate='3321 AR',
-        email='[email protected]',
-        phone_number='0539003545',
-        driver_id='ID2220000000'
-    )
-
-    # Add records to the session
-    session.add(vehicle_data_1)
-    session.add(vehicle_data_2)
-
-    # Commit the records to the database
-    session.commit()
-
-    # Query the table to confirm data
-    vehicles = session.query(VehicleInformation).all()
-
-    # Close the session
-    session.close()
 
 # Streamlit app main function
 def main():