build chain

app.py

@@ -81,13 +81,24 @@ def main():
     if st.button("classify KBOB!"):
         if uploaded_file is not None:
             with st.spinner("GPT4 at work"):
-                my_prompt = get_pdf_text(uploaded_file)
-                answer = gpt4_new(my_prompt)
-
-                #process_prompts_and_save(prompts)
+                pdf_text = str(get_pdf_text(uploaded_file))
+                prompt_1 = auftrag_1a + auftrag_1b + 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_1a + auftrag_1c + 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 =
+                #answer_3 =
+                #print(prompt_2)
             st.success("work done, saved")
-            st.write(answer)
-            st.download_button("download txt", json_open("gpt4_responses.txt"), file_name="gpt4_responses.txt")
+
+            #st.download_button("download txt", json_open("gpt4_responses.txt"), file_name="gpt4_responses.txt")
         else:
             st.warning("no file")
 
@@ -96,9 +107,11 @@ if __name__ == "__main__":
     #prompts = ["classify the document, tell me the ", "hello"]
     #process_prompts_and_save(prompts)
     auftrag_1a = "Klassifiziere dieses Dokument nach "
+    auftrag_1b = "diesen 'Baubranchen Disziplinen': "
+    auftrag_1c = "diesen 'Dokumententypen': "
     Baubranchen_Disziplinen = ['Lüftung', 'Heizung', 'Sanitär', 'Elektro', 'Architektur', 'Statik']
-    auftrag_1b = "gib nur den am besten passendsten Eintrag zurück. Dokumenteninhalt: "
+    auftrag_2 = "gib nur den am besten passendsten Eintrag zurück. Dokumenteninhalt: "
     Dokumententypen = ['Fotodokumentation', 'Projektdokumentation (PD)', 'Objektdokumentation (OD)',
                        'Prozessdokumentation',  'Fachdokumentation', 'Anlagedokumentation']
-
+    #print(str(Baubranchen_Disziplinen))
     main()