Update app.py

app.py

@@ -20,15 +20,15 @@ import torch
 
 
 # Zugriff auf das Secret als Umgebungsvariable
-HF_WRITE = os.getenv("HF_WRITE")
+#HF_WRITE = os.getenv("HF_WRITE")
 HF_READ = os.getenv("HF_READ")
 
 # CONSTANTS
-REPO_ID = "alexkueck/kkg_suche"
-REPO_TYPE = "space"
+#REPO_ID = "alexkueck/kkg_suche"
+#REPO_TYPE = "space"
 
 # HfApi-Instanz erstellen
-api = HfApi()
+#api = HfApi()
 
 
 # Relativer Pfad zum Verzeichnis mit den Dokumenten
@@ -40,7 +40,9 @@ DOCS_DIR = "kkg_dokumente"
 model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
 
 
-# Funktion zum Extrahieren von Text aus PDF
+########################################################
+######## Hilfsfunktionen ###############################
+# Funktion zum Extrahieren von Text aus PDF - und Word
 def extract_text_from_pdf(pdf_path):
     doc = fitz.open(pdf_path)
     pages = []
@@ -70,7 +72,8 @@ def extract_text_from_docx(docx_path):
         pages.append({'header': header, 'content': '\n'.join(current_page)})
     return pages
 
-# Initialisierung der Dokumente
+
+# Initialisierung der Dokumente - Dictionary um die Dokuemtneteninhalte, Switen und Überschriften zu halten
 def initialize_documents():
     documents = []
     if os.path.exists(DOCS_DIR):
@@ -85,6 +88,21 @@ def initialize_documents():
                 documents.append({"file": file_name, "pages": pages})
     return documents
 
+
+######################################
+#nach relevanten suche -> download Link der passenden Dokuemtne erstellen
+def download_link(doc_name):
+    # URL für das Herunterladen der Datei
+    file_url = f"https://huggingface.co/spaces/alexkueck/kkg_suche/resolve/main/kkg_dokumente/{doc_name}?token={HF_READ}"
+    return f'<b><a href="{file_url}" target="_blank" style="color: #BB70FC; font-weight: bold;">{doc_name}</a></b>'
+
+
+# Zeitelumbrüche entfernen
+def remove_line_breaks(text):
+    return text.replace('\n', ' ').replace('\r', ' ')
+
+########################################################
+######## Hilfsfunktionen Datei-Upload ##################
 # Hochladen von Dateien
 def upload_pdf(file):
     if file is None:
@@ -113,25 +131,18 @@ def display_files():
         file_size = os.path.getsize(file_path) / 1024  # Größe in KB
         row_color = "#4f4f4f" if i % 2 == 0 else "#3a3a3a"  # Wechselnde Zeilenfarben
         files_table += f"<tr style='background-color: {row_color}; border-bottom: 1px solid #ddd;'>"
-        files_table += f"<td>{download_link(file)}</td>"
+        files_table += f"<td><b>{download_link(file)}</b></td>"
         files_table += f"<td>{file_size:.2f}</td></tr>"
     files_table += "</table>"
     return files_table
 
 
+# gefundene relevante Dokumente auflisten (links)
 def list_pdfs():
     if not os.path.exists(DOCS_DIR):
         return []
     return [f for f in os.listdir(SAVE_DIR) if f.endswith('.pdf')]
 
-######################################
-#nach relevanten suche -> download Link der passenden Dokuemtne erstellen
-def download_link(doc_name):
-    # URL für das Herunterladen der Datei
-    file_url = f"https://huggingface.co/spaces/alexkueck/kkg_suche/resolve/main/kkg_dokumente/{doc_name}?token={HF_READ}"
-    return f'<b><a href="{file_url}" target="_blank" style="color: #b56edc; font-weight: bold;">{doc_name}</a></b>'
-
-
 
 ###########################################################
 ############# KI um Suchanfrage zu Embedden ###############
@@ -202,15 +213,12 @@ def search_documents(query):
                 if header_content:
                     snippet += f"<b>Überschrift:{header_content}</b> <br>"
                     
-                snippet += f"{page_content[start:end]}<br><hr>"
+                snippet += f"{remove_line_breaks(page_content[start:end])}<br><hr>"
                 relevant_text += snippet
 
                 if doc['file'] not in relevant_docs:
                     relevant_docs[doc['file']] = []
                 relevant_docs[doc['file']].append(snippet)
-                
-
-    
 
     # Sortieren nach Relevanz
     results = sorted(results, key=lambda x: x[1], reverse=True)
@@ -220,7 +228,8 @@ def search_documents(query):
     return results, relevant_text
 
 
-
+###########################################################
+############## Vorbereitung View in gradio ################
 #######################################
 #Suche starten und View aktialisieren
 def search_and_update(query):
@@ -240,6 +249,7 @@ def search_and_update(query):
         relevant_text = "<div class='no-results'>Kein relevanter Text gefunden.</div>"
     
     return doc_links, relevant_text
+    
 
 ######################################################################
 ############### Anwendung starten ####################################
@@ -257,7 +267,7 @@ with gr.Blocks(css="""
     }
     .doc-name {
         font-weight: bold;
-        color: #B05DF9; /* Helles Lila für verlinkte Dokumente */
+        color: #B05DF9; /* Dunkleres Lila für verlinkte Dokumente */
     }
     .page-number {
         font-weight: bold;
@@ -273,7 +283,7 @@ with gr.Blocks(css="""
         width: 100%; /* Volle Breite */
     }
     #doc_links a {
-        color: #B05DF9; /* Helles Lila für Links im doc_links Feld */
+        color: #BB70FC; /* Helles Lila für Links im doc_links Feld */
         font-weight: bold;
         width: 100%; /* Volle Breite */
     }