added so default descriptions runs if no logfile

Sanjivani-S · May 6, 2024 · 7e09c7a · 7e09c7a
1 parent 82d07f0
commit 7e09c7a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test_draftgpt.py b/test_draftgpt.py
@@ -58,9 +58,12 @@ def draft_gpt(user_input, openai_api_key=os.environ["OPENAI_API_KEY"], gpt_model
             incident_desc = response.text
         else:
             print("Failed to fetch file:", response.status_code)
-    else:
+    elif input_logfile is not None:
         with open(input_logfile, "r") as file:
-            incident_desc = file.read()
+            incident_desc = file.read().strip()
+    else:
+        with open("incident_descriptions/incident_description.txt", "r") as file:
+            incident_desc = file.read().strip()
 
     print("\n contents of file read == \n")   
     print(incident_desc)