Update app.py

app.py

@@ -432,11 +432,20 @@ def gpt_answer(api_key, query, model="gpt-3.5-turbo-1106", system_prompt="Use th
 
 def add_line_breaks(input_string, line_length=100):
     lines = []
-    input_string=input_string.split("\n---\n[Sources]")[0]
-    for i in range(0, len(input_string), line_length):
-        line = input_string[i:i+line_length]
+    to_break=input_string.split("\n---\n[Sources]")[0]
+    for i in range(0, len(to_break), line_length):
+        line = to_break[i:i+line_length]
         lines.append(line)
-    return '\n'.join(lines)
+    return '\n'.join(lines)+input_string[len(to_break)-1:]
+
+
+
+def upload_text_file(content):
+    data = {"content": content, "syntax": "text", "expiry_days": 1}
+    headers = {"User-Agent": "Sources"}
+    r = requests.post("https://dpaste.com/api/", data=data, headers=headers)
+    return r.text + ".txt"
+
 
 def ask_df(df, api_key, model, ui_session_id):
     answers = []
@@ -454,11 +463,8 @@ def ask_df(df, api_key, model, ui_session_id):
         docs = db.similarity_search(question)
         references = '\n******************************\n'.join([d.page_content for d in docs])
         print(f"REFERENCES: {references}")
-        sources_file = f"{secrets.token_urlsafe(16)}.txt"
-        with open(sources_file, 'w') as file:
-            file.write(references)
         try:
-            source = f"https://organizedprogrammers-docgpt-table.hf.space/file={sources_file}"
+            source = upload_text_file(references)
         except:
             source = "ERROR WHILE GETTING THE SOURCES FILE"
         query = f"## USER QUESTION:\n{question}\n\n## REFERENCES:\n{references}\n\nANSWER:\n\n"