downloadable classification

app.py

@@ -100,20 +100,26 @@ def main():
                     prompt_1 = auftrag_0 + auftrag_1_disz + str(Baubranchen_Disziplinen) + pdf_text
                     answer_1 = gpt4_new(prompt_1)
                     print(prompt_1)
+
                 st.write(answer_1)
                 with st.spinner("GPT4 at work"):
                     prompt_2 = auftrag_0 + auftrag_1_dokt + str(Dokumententypen) + pdf_text
                     answer_2 = gpt4_new(prompt_2)
                     print(prompt_2)
+
                 st.write(answer_2)
                 with st.spinner("GPT4 at work"):
                     prompt_3 = auftrag_0 + auftrag_1_gesch + str(ASH_Geschosse) + pdf_text
                     answer_3 = gpt4_new(prompt_3)
                     print(prompt_3)
+                    with open('gpt4_responses.txt', 'w', encoding='utf-8') as file:
+                        file.writelines(answer_1)
+                        file.writelines(answer_2)
+                        file.writelines(answer_3)
                 st.write(answer_3)
                 st.success("work done, saved")
 
-                st.download_button("download txt")
+                st.download_button("download txt", json_open("gpt4_responses.txt"), file_name="classification.txt")
                 #st.download_button("download txt", json_open("gpt4_responses.txt"), file_name="gpt4_responses.txt")
             else:
                 st.warning("no file")