Update app.py

app.py

@@ -17,67 +17,10 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
 logger = logging.getLogger(__name__)
 
 # Email and database configuration
-DB_CONFIG = {
-    'host': '0.tcp.in.ngrok.io',
-    'port': 13890,
-    'user': 'root',
-    'password': '',
-    'database': 'shipment_details'
-}
-
-output_format = {
-    "origin": None,
-    "destination": None,
-    "expected_shipment_datetime": None,
-    "types_of_service": None,
-    "warehouse": None,
-    "description": None,
-    "quantities": None,
-    "carrier_details": None
-}
+DB_CONFIG = os.getenv('db')
 
 # System prompt for LLM
-prompt = f"""
-System prompt: You will be provided with an email containing shipment details. Your task is to extract specific information based on the given instructions.
-
-Instructions:
-1. The input email may contain irrelevant information. Focus only on extracting details about future shipments.
-2. The output should be in JSON format with all values to use only double quotes. If a type of information is not found, it should be marked as null.
-3. Extract the following information:
-    - origin: The origin location of the consignment.
-    - destination: The destination location of the consignment.
-    - expected_shipment_datetime: The expected date and time of delivery to the warehouse (format: yyyy-mm-dd hh:mm:ss).
-    - types_of_service: The type of service (AIR, LCL, FCL).
-    - warehouse: The name of the warehouse.
-    - description: A brief description of the email (ASN).
-    - quantities: The number of items in the shipment.
-    - carrier_details: The details of the carrier.
-4. The output extracted information must be in this format:
-{{
-    "origin": "",
-    "destination": "",
-    "expected_shipment_datetime": "",
-    "types_of_service": "",
-    "warehouse": "",
-    "description": "",
-    "quantities": "",
-    "carrier_details": ""
-}}
-
-Examples:
-1. Email: We are pleased to inform you of an upcoming shipment originating from Hamburg...
-   Extracted Information:
-        origin: Hamburg,
-        destination: New York,
-        expected_shipment_datetime: 2024-08-15 00:00:00,
-        types_of_service: AIR,
-        warehouse: Sky Logistics,
-        description: We are pleased to inform you of an upcoming shipment...
-        quantities: 200 units,
-        carrier_details: Sky Logistics
-
-Output: {output_format}
-"""
+prompt = os.getenv('prompt')
 
 # Function to insert extracted shipment details into MySQL database
 def insert_data(extracted_details):