geplante Erweiterungen

ai_generated_metadata.txt

@@ -1,10 +1,6 @@
 
-ISB-020-U3-W-E-01-B15100-001-000.pdf;L-Lüftung;'Anlagedokumentation';'Anlagedokumentation';
+ISB-020-U3-W-E-01-B15100-006-000.pdf;L-Lüftung;'Anlagedokumentation';'Anlagedokumentation';
 
-ISB-020-U3-W-E-01-B15100-002-000.pdf;L-Lüftung;'Prozessdokumentation';'Prozessdokumentation';
+ISB-020-U3-W-H-01-O15004-002-000.pdf;H-Heizung;'Anlagedokumentation';'Anlagedokumentation';
 
-ISB-020-U3-W-E-01-B15100-003-000.pdf;L-Lüftung;'Anlagedokumentation';'Anlagedokumentation';
-
-ISB-020-U3-W-E-01-B15100-004-000.pdf;L-Lüftung;'Anlagedokumentation';'Anlagedokumentation';
-
-ISB-020-U3-W-E-01-B15100-005-000.pdf;L-Lüftung;'Prozessdokumentation';'Prozessdokumentation';
+ISB-020-U3-W-E-01-B07005-001-040.pdf;E-Elektroanlagen;<no classification>;<no classification>;

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import os
-import openai
+# import openai
 from PyPDF2 import PdfReader
 from openai import OpenAI
 from langchain.chat_models import ChatOpenAI
@@ -26,6 +26,7 @@ def gpt4_new(prompt_text):
 
 # Define a function to ask a question to GPT-4
 def ask_gpt4(question):
+    print(question)  # we don't have to submit the question?
     try:
         # Use the chat function to send a message and get a response
         response = ChatOpenAI()
@@ -83,11 +84,18 @@ def json_open(filename):
 
 def main():
     st.title("Doc Classifier")
-    st.subheader("Limitationen: ")
-    st.write("bisher nur PDFs")
-    st.write("nur Disziplin, Doc typ. und Geschoss")
-    st.write("macht noch viele Fehler, vor allem bei Koordination, Datennetz usw, (unklare Disziplinen)")
-    st.write("")
+    l, r = st.columns(2)
+    with l:
+        st.subheader("Limitationen: ")
+        st.write("bisher nur PDFs")
+        st.write("nur Disziplin, Doc typ. und Geschoss")
+        st.write("macht noch viele Fehler, vor allem bei Koordination, Datennetz usw, (unklare Disziplinen)")
+        st.write("")
+    with r:
+        st.subheader("geplante Erweiterungen:")
+        st.write("text beschreibung")
+        st.write("jpg, bilder, tabellen, .xlsx, .docx")
+        st.write("Ecodomus API einbinden")
 
     if st.text_input("ASK_ASH_PASSWORD: ", type="password") == ASK_ASH_PASSWORD:
         uploaded_files = st.file_uploader("PDF Dokument", accept_multiple_files=True)
@@ -104,10 +112,10 @@ def main():
                         st.write("Disziplin")
                         st.write(f"")
                     with col2:
-                        st.write("Disziplin")
+                        st.write("Dokumententyp")
                         st.write(f"")
                     with col3:
-                        st.write("Disziplin")
+                        st.write("Geschoss")
                         st.write(f"")
 
                     for file in uploaded_files:
@@ -115,7 +123,7 @@ def main():
                         with col1:
                             with st.spinner("GPT4 at work"):
                                 pdf_text = str(get_pdf_text(file))
-                                prompt_1 = auftrag_0 + auftrag_1_disz + str(Baubranchen_Disziplinen) + pdf_text
+                                prompt_1 = auftrag_0 + auftrag_1_disziplin + str(Baubranchen_Disziplinen) + pdf_text
                                 answer_1 = gpt4_new(prompt_1)
                                 print(prompt_1)
                                 metadata.append(answer_1)
@@ -123,7 +131,7 @@ def main():
 
                         with col2:
                             with st.spinner("GPT4 at work"):
-                                prompt_2 = auftrag_0 + auftrag_1_dokt + str(Dokumententypen) + pdf_text
+                                prompt_2 = auftrag_0 + auftrag_1_type + str(Dokumententypen) + pdf_text
                                 answer_2 = gpt4_new(prompt_2)
                                 print(prompt_2)
                                 metadata.append(answer_2)
@@ -132,7 +140,7 @@ def main():
 
                         with col3:
                             with st.spinner("GPT4 at work"):
-                                prompt_3 = auftrag_0 + auftrag_1_gesch + str(ASH_Geschosse) + pdf_text
+                                prompt_3 = auftrag_0 + auftrag_1_ge + str(ASH_Geschosse) + pdf_text
                                 answer_3 = gpt4_new(prompt_3)
                                 print(prompt_3)
                                 metadata.append(answer_2)
@@ -161,9 +169,9 @@ if __name__ == "__main__":
     #prompts = ["classify the document, tell me the ", "hello"]
     #process_prompts_and_save(prompts)
     auftrag_0 = "Klassifiziere dieses Dokument nach "
-    auftrag_1_disz = "diesen 'Baubranchen Disziplinen': "
-    auftrag_1_dokt = "diesen 'Dokumententypen': "
-    auftrag_1_gesch = "diesen 'Geschossen': "
+    auftrag_1_disziplin = "diesen 'Baubranchen Disziplinen': "
+    auftrag_1_type = "diesen 'Dokumententypen': "
+    auftrag_1_ge = "diesen 'Geschossen': "
     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',