Improved source view.3

app.py

@@ -194,7 +194,7 @@ if prompt := st.chat_input("Ihre Frage eingeben..."):
             
             message_placeholder.markdown(full_response)
             
-            # G) Quellen Anzeige (Hierarchie-Shift: Quelle wird Haupt-Header)
+            # G) Quellen Anzeige (Hierarchie-Shift + Keyword Highlighting)
             st.markdown("---")
             st.markdown(f"**Top-Quellen (Gefiltert aus 100):**")
             st.markdown(", ".join(list(unique_sources_labels)))
@@ -202,29 +202,39 @@ if prompt := st.chat_input("Ihre Frage eingeben..."):
             with st.expander("Details anzeigen"):
                 st.caption(f"Modell: {model_name} | Sortiert durch FlashRank")
                 
+                # Suchbegriffe für das Highlighting vorbereiten (kleine Wörter ignorieren)
+                search_terms = [word.strip(",.?!") for word in prompt.split() if len(word) > 3]
+
                 for i, item in enumerate(source_details):
                     # 1. Die Quelle selbst wird nun zum obersten Header (#)
                     st.markdown(f"# {item['label']}")
                     st.caption(f"Relevanz-Score: {item['score']:.4f}")
                     
-                    # 2. FLATTENING: Wir verschieben alle Header im Originaltext um eine Ebene nach unten
-                    # WICHTIG: Wir ersetzen von groß nach klein, um Doppel-Ersetzungen zu vermeiden
+                    # 2. HIERARCHIE-SHIFT: Alle Header um eine Ebene nach unten
                     text = item['text']
-                    text = text.replace("#####", "######")
-                    text = text.replace("####", "#####")
-                    text = text.replace("###", "####")
-                    text = text.replace("##", "###")
-                    # Der ursprüngliche Haupt-Header wird nun zum Sub-Header
-                    flattened_text = text.replace("# ", "## ") 
+                    # Von klein nach groß ersetzen, um Kaskaden-Effekte zu vermeiden
+                    for level in range(5, 0, -1):
+                        old_prefix = "#" * level + " "
+                        new_prefix = "#" * (level + 1) + " "
+                        text = text.replace(old_prefix, new_prefix)
+                    
+                    # 3. KEYWORD HIGHLIGHTING (Optionaler Bonus)
+                    # Wir ersetzen die Begriffe im Text durch eine fette Version
+                    highlighted_text = text
+                    for term in search_terms:
+                        # Case-insensitive Ersetzung für die Anzeige
+                        import re
+                        pattern = re.compile(re.escape(term), re.IGNORECASE)
+                        highlighted_text = pattern.sub(f"**\g<0>**", highlighted_text)
                     
-                    # 3. Text ausgeben
-                    st.markdown(flattened_text)
+                    # 4. Text ausgeben
+                    st.markdown(highlighted_text)
                     
-                    # 4. Saubere Trennung durch eine Linie
+                    # 5. Saubere Trennung
                     if i < len(source_details) - 1:
                         st.markdown("---")
 
             st.session_state.messages.append({"role": "assistant", "content": full_response})
-            
+
         except Exception as e:
             st.error(f"Fehler: {e}")