Update app.py

app.py

@@ -183,12 +183,15 @@ def read_email():
         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)
-    
+        if len(t)==o:
+            print('input : ',i)
+            print('output : ',t)
+            # t = t[t.find('json\n')+len('json\n'):]
+            extracted_details = json.loads(t)
+        else:
+            continue
+            
+        print(extracted_details
         # print(extracted_details)
         print(type(extracted_details))
         meta_data = {