Update app.py

app.py

@@ -429,6 +429,20 @@ def gpt_answer(api_key, query, model="gpt-3.5-turbo-1106", system_prompt="Use th
     )
     return chat_completion.choices[0].message.content
 
+
+def add_line_breaks(input_string, line_length=60):
+    lines = []
+
+    for i in range(0, len(input_string), line_length):
+        line = input_string[i:i+line_length]
+        if '\n' not in line:
+            lines.append(line)
+        else:
+            lines.append(line.split('\n', 1)[0])
+            lines.append('')
+    
+    return '\n'.join(lines)
+
 def ask_df(df, api_key, model, ui_session_id):
     answers = []
     session_folder = f"PDFAISS-{ui_session_id}"
@@ -458,7 +472,7 @@ def ask_df(df, api_key, model, ui_session_id):
         except Exception as e:
             answer = "ERROR WHILE ANSWERING THE QUESTION"
             print("ERROR: ", e)
-        complete_answer = "\n---\n".join(["## " + question, answer, "[Sources](" + source + ")"])
+        complete_answer = add_line_breaks("\n---\n".join(["## " + question, answer, "[Sources](" + source + ")"]))
         answers.append(complete_answer)
         print(complete_answer)
     df[question_column] = answers