Update app.py

app.py

@@ -177,38 +177,33 @@ def insert_data(extracted_details):
 llm = Llama(model_path='./ggml-model-q8_0.gguf', n_ctx=2048, n_batch=2048, n_threads= 204)
 
 def read_email():
-    try:
-        df = pd.read_csv('./emails.csv')
-        for i in df['Body']:
-            prompt_ = f"<s>system:{prompt}<|end|><s>user:{i}<|end|<s>assistant:"
-            output = llm(prompt_, max_tokens=1000, temperature=0.1)
-            print("*"*50)
-            t = output['choices'][0]['text']
-            print('input : ',i)
-            print('output : ',t)
-            # t = t[t.find('json\n')+len('json\n'):]
-            extracted_details = json.loads(t)
-            print(extracted_details)
-
-            # print(extracted_details)
-            print(type(extracted_details))
-            meta_data = {
-                'sender': 'sender',
-                'receiver': 'receiver',
-                'cc': 'cc',
-                'bcc': 'bcc',
-                'subject': 'subject'
-            }
-            # print(type(meta_data))
-            extracted_details.update(meta_data)
+    df = pd.read_csv('./emails.csv')
+    for i in df['Body']:
+        prompt_ = f"<s>system:{prompt}<|end|><s>user:{i}<|end|<s>assistant:"
+        output = llm(prompt_, max_tokens=1000, temperature=0.1)
+        print("*"*50)
+        t = output['choices'][0]['text']
+        print('input : ',i)
+        print('output : ',t)
+        # t = t[t.find('json\n')+len('json\n'):]
+        extracted_details = json.loads(t)
+        print(extracted_details)
+    
+        # print(extracted_details)
+        print(type(extracted_details))
+        meta_data = {
+            'sender': 'sender',
+            'receiver': 'receiver',
+            'cc': 'cc',
+            'bcc': 'bcc',
+            'subject': 'subject'
+        }
+        # print(type(meta_data))
+        extracted_details.update(meta_data)
     
-            print('full data about email ! ...::',extracted_details)
-            insert_data(extracted_details)
+        print('full data about email ! ...::',extracted_details)
+        insert_data(extracted_details)
     
-        
-    except Exception as e:
-        print(f"Error reading csv emails: {e}")
-
 
 # Global variables
 running = False