build chain II, password, explained limitations

app.py

@@ -11,7 +11,10 @@ def gpt4_new(prompt_text):
     response = client.chat.completions.create(
         model="gpt-4",
         messages=[{"role": "system",
-                   "content":   "Du bist eine Maschine, die Dokumente klassifiziert."},
+                   "content":   "Du bist eine Maschine, die Dokumente klassifiziert."
+                                "Du gibts möglichst kurze Antworten, am besten maschnienlesbar"
+                                "und nur ein einziges Wort. Du gibts keine Erklärungen oder"
+                                "Begründungen."},
                   {"role": "user", "content": prompt_text}])
     return response.choices[0].message.content
 
@@ -74,33 +77,42 @@ def json_open(filename):
 
 
 def main():
+    ASK_ASH_PASSWORD = os.environ["ASK_ASH_PASSWORD"]
     st.title("Doc Classifier")
-    uploaded_file = st.file_uploader("files", accept_multiple_files=False)
-    #print(uploaded_file)
-    #print(uploaded_file.name)
-    if st.button("classify KBOB!"):
-        if uploaded_file is not None:
-            with st.spinner("GPT4 at work"):
-                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.download_button("download txt", json_open("gpt4_responses.txt"), file_name="gpt4_responses.txt")
-        else:
-            st.warning("no file")
+    st.subheader("Limitationen: ")
+    st.write("bisher nur PDFs")
+    st.write("nur Disziplin und Doc typ.")
+    st.write("Dokumenten Grösse ist stark limitiert")
+    st.write("macht noch viele Fehler, vor allem bei Koordination, Datennetz usw, (unklare Disziplinen)")
+    st.write("")
+
+    if st.text_input("ASK_ASH_PASSWORD: ", type="password") == ASK_ASH_PASSWORD:
+        uploaded_file = st.file_uploader("PDF Dokument", accept_multiple_files=False)
+        #print(uploaded_file)
+        #print(uploaded_file.name)
+        if st.button("classify KBOB!"):
+            if uploaded_file is not None:
+                with st.spinner("GPT4 at work"):
+                    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.download_button("download txt", json_open("gpt4_responses.txt"), file_name="gpt4_responses.txt")
+            else:
+                st.warning("no file")
 
 
 if __name__ == "__main__":
@@ -109,8 +121,15 @@ if __name__ == "__main__":
     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_2 = "gib nur den am besten passendsten Eintrag zurück. Dokumenteninhalt: "
+    Baubranchen_Disziplinen = ['A-Architektur', 'B-Bauphysik', 'C-Rohrpostanlagen', 'D-Datennetz', 'E-Elektroanlagen',
+                               'F-Fassadenplanung', 'G-Küche', 'H-Heizung', 'I-Innenausbau', 'K-Kälte', 'L-Lüftung',
+                               'M-Medizintechnik', 'N-Fördertechnik', 'O-Gebäudebetrieb', 'P-Sprinkler',
+                               'Q-Brandschutz', 'R-Koordination', 'S-Sanitär', 'T-Tragwerksplanung', 'W-Informatik',
+                               'Z-Lichtplanung']
+    auftrag_2 = "gib nur den am besten passendsten Eintrag zurück. " \
+                "Keine weiteren Ausführungen oder Erklärungen. " \
+                "Antworte am besten in einem Wort. " \
+                "Hier der Dokumenteninhalt: "
     Dokumententypen = ['Fotodokumentation', 'Projektdokumentation (PD)', 'Objektdokumentation (OD)',
                        'Prozessdokumentation',  'Fachdokumentation', 'Anlagedokumentation']
     #print(str(Baubranchen_Disziplinen))