Update utils.py

utils.py

@@ -104,20 +104,34 @@ def summary(input_json):
     output["Stock Ticker"] = input_json['symbol']
 
     answer = callAzure(promptShort[id], long_text)
-    idx = answer.index("\n")
+    try:
+        idx = answer.index("\n")
+    except:
+        idx = 0
     output['Short Summary'] = answer[idx+2:]
 
     answer = callAzure(promptLong[id], long_text)
-    idx = answer.index("\n")
+    try:
+        idx = answer.index("\n")
+    except:
+        idx = 0
     output['Long summary'] = answer[idx+2:]
 
     prompt = "1 word Financial SEO tag for this news article"
     answer = callAzure(prompt, output['Short Summary'])
-    idx = answer.index("\n")
+    try:
+        idx = answer.index("\n")
+    except:
+        idx = 0
     output['Tag'] = answer[idx+2:]
 
     prompt = "Give a single headline for this News Article"
-    output['Headline'] = callAzure(prompt, output['Short Summary'])
+    answer = callAzure(prompt, output['Short Summary'])
+    try:
+        idx = answer.index("\n")
+    except:
+        idx = 0
+    output['Headline'] = answer[idx+2:]
 
     utc_now = datetime.datetime.utcnow()
     ist_now = utc_now.astimezone(datetime.timezone(datetime.timedelta(hours=5, minutes=30)))