Update LLM_automation_Groq.py

LLM_automation_Groq.py

@@ -104,18 +104,38 @@ def create_data(description):
     for i in range(len(dj2)):
         words = dj2[i].split(";")  # Splitting at the semicolon delimiter
         #print(f"Date: {words[0]}")
-        Date.append(words[0])
+        try:
+            Date.append(words[0])
+        except:
+            Date.append("Could not extract date")
             
         #print(f"Time: {words[1]}")
-        Time.append(words[1])
-            
+        try:
+            Time.append(words[1])
+        except:
+            Time.append("Could not extract time")
+    
         #print(f"Casualities: {words[2]}")
-        Killed.append(words[2])
-        Injured.append(words[3])
-        Location.append(words[4])
-        Road_Characteristic.append(words[5])
-        Pedestrian_Involved.append(words[6])
-        #Vehicles_involved.append(words[7])
+        try:
+            Killed.append(words[2])
+        except:
+            Killed.append("Could not extract killed")
+        try: 
+            Injured.append(words[3])
+        except:
+            Injured.append("Could not extract injured")
+        try:
+            Location.append(words[4])
+        except:
+            Location.append("Could not extract location")
+        try:
+            Road_Characteristic.append(words[5])
+        except:
+            Road_Characteristic.append("Could not extract road characteristic")
+        try:
+            Pedestrian_Involved.append(words[6])
+        except:
+            Pedestrian_Involved.append("Could not extract pedestrian involved")
 
     #### Probable type of final dataframe:
     df2["Accident Date"]=Date