import streamlit as st
import requests
from bs4 import BeautifulSoup
from sentence_transformers import SentenceTransformer, util
from itertools import combinations
import numpy as np

# Load an embedding model
model = SentenceTransformer('mixedbread-ai/deepset-mxbai-embed-de-large-v1')

# Ideal answer for comparison
ideal_answer = """
Das Weihnachtsfest in Deutschland ist geprägt von einer Vielzahl an Bräuchen, Symbolen und Traditionen, die regional unterschiedliche Ausprägungen haben.
Neben allgemeinen, landesweiten Traditionen existieren in den einzelnen Regionen eigene Bräuche, die das Weihnachtsfest besonders machen.

Allgemeine Bräuche und Symbole
Zu den bekanntesten Symbolen und Bräuchen, die landesweit verbreitet sind, zählen der Adventskranz, der Weihnachtsbaum, der Nikolaustag, der Weihnachtsmarkt sowie bestimmte kulinarische Spezialitäten.
Diese Traditionen schaffen eine festliche Atmosphäre und symbolisieren für viele Deutsche die Weihnachtszeit.

Advent und Adventskranz: Der Advent, die Vorbereitungszeit auf Weihnachten, beginnt am vierten Sonntag vor Weihnachten.
In dieser Zeit wird oft ein Adventskranz mit vier Kerzen aufgestellt, von denen jeden Sonntag eine weitere angezündet wird.
Dieser Brauch ist besonders im ganzen Land verbreitet und symbolisiert die wachsende Vorfreude auf das Fest.

Weihnachtsbaum: Der Weihnachtsbaum ist das zentrale Symbol des Weihnachtsfestes in Deutschland.
Meistens wird er am Heiligen Abend aufgestellt und mit Kerzen, Kugeln und manchmal auch selbstgebasteltem Schmuck dekoriert.
Der Brauch stammt ursprünglich aus dem Elsass und verbreitete sich im 19. Jahrhundert über ganz Deutschland.

Nikolaustag: Am 6. Dezember wird der Nikolaustag gefeiert. Kinder stellen am Abend vorher ihre Stiefel vor die Tür, die über Nacht vom „Nikolaus“ mit Süßigkeiten und kleinen Geschenken gefüllt werden.
Dieser Brauch ist in vielen Regionen verbreitet, zeigt jedoch in einigen Gebieten, besonders in Süddeutschland, Abweichungen, da dort oft der „Knecht Ruprecht“ oder „Krampus“ den Nikolaus begleitet.

Weihnachtsmärkte: Weihnachtsmärkte sind in Deutschland besonders beliebt und gehören zur Adventszeit.
Sie bieten regionale Spezialitäten, Weihnachtsdekoration und handgemachte Geschenke. Bekannte Weihnachtsmärkte gibt es in Städten wie Nürnberg, Dresden und München.
Diese Märkte variieren in ihrem Angebot und Charakter je nach Region, was besonders für Touristen attraktiv ist.

Weihnachtsessen: Traditionell wird am Heiligen Abend oft eine einfache Mahlzeit wie Kartoffelsalat mit Würstchen serviert, während das Festessen am ersten Weihnachtstag stattfindet.
Gänsebraten, Rotkohl und Klöße sind typische Speisen für das Weihnachtsessen. Auch hier gibt es regionale Unterschiede: Im Süden Deutschlands werden oft Bratwürste serviert, während im Norden Fischgerichte beliebter sind.

Regionale Unterschiede
Die regionalen Unterschiede beim Weihnachtsfest in Deutschland zeigen sich nicht nur in der Art der Bräuche, sondern auch in speziellen Traditionen:

Bayern und der Alpenraum: In Bayern und im Alpenraum gibt es besonders reichhaltige Bräuche. Hier ist das „Krippenspiel“ weit verbreitet, bei dem die Geburt Jesu in einem kleinen Theaterstück nachgestellt wird.
In vielen Dörfern und Städten findet auch das „Christkindl-Anschießen“ statt, bei dem Schützenvereine am Heiligen Abend Böllerschüsse abfeuern, um das Christkind willkommen zu heißen.

Norddeutschland: In Norddeutschland sind Weihnachtslieder und das Singen von Shanties, also traditionellen Seemannsliedern, häufig Teil der Festlichkeiten.
Auch die Dekoration unterscheidet sich leicht, mit einem stärkeren Fokus auf maritime Symbole. Der Weihnachtsmann braven Kindern am Heiligen Abend Geschenke.

Sachsen und das Erzgebirge: Sachsen, insbesondere das Erzgebirge, ist bekannt für seinen reich verzierten Weihnachtsschmuck und das Kunsthandwerk.
Schwibbögen, Räuchermännchen und Nussknacker sind Symbole dieser Region und stammen ursprünglich aus der Bergbaugeschichte des Erzgebirges, in der Licht eine besondere Rolle spielte.

Rheinland: Im Rheinland ist die Tradition des Sternsingens weit verbreitet. Kinder ziehen als „Heilige Drei Könige“ verkleidet von Haus zu Haus und singen Lieder, um Spenden für wohltätige Zwecke zu sammeln.
Diese Tradition wird zwar auch in anderen Teilen Deutschlands gepflegt, hat jedoch im Rheinland eine besonders lange Geschichte.
"""

def fetch_wikipedia_content(url):
    try:
        response = requests.get(url)
        soup = BeautifulSoup(response.text, 'html.parser')
        paragraphs = soup.find_all('p')
        content = ' '.join([para.text for para in paragraphs])
        return content
    except Exception as e:
        st.error(f"Error fetching article: {e}")
        return None

def evaluate_contexts(student_texts):
    ideal_embedding = model.encode(ideal_answer, convert_to_tensor=True)
    scores = {"relevance": [], "completeness": 0, "depth": [], "regional_completeness": None}

    # Define keywords and region-specific terms
    keywords = ["Advent", "Adventskranz", "Weihnachtsbaum", "Nikolaus", "Weihnachtsmarkt", "Weihnachtsessen",
                "Weihnachtslieder", "Weihnachtsschmuck", "Krippenspiel", "Weihnachtsgebäck", "Bescherung", "Wichteln"]
    region_keywords = {
        "Süddeutschland": ["Christkind", "Krampus", "Böllerschüsse"],
        "Norddeutschland": ["Weihnachtsmann", "Weihnachtslieder"],
        "Ostdeutschland": ["Schwibbogen", "Weihnachtspyramide", "Nussknacker"],
        "Westdeutschland": ["Sternsingen", "Adventskalender"]
    }

    all_found_keywords = set()
    regional_keyword_coverage = {region: 0 for region in region_keywords}
    combined_text = " ".join(student_texts)

    for text in student_texts:
        student_embedding = model.encode(text, convert_to_tensor=True)
        relevance_score = util.cos_sim(student_embedding, ideal_embedding).item()
        scores["relevance"].append(relevance_score)

        found_keywords = [keyword for keyword in keywords if keyword in text]
        unique_keyword_count = len(set(found_keywords))
        all_found_keywords.update(found_keywords)

    # Calculate overall scores
    scores["relevance"] = np.mean(scores["relevance"])
    scores["completeness"] = len(all_found_keywords) / len(keywords)
    scores["depth"] = np.mean(scores["depth"])

    # Regional completeness: Check for region-specific keyword coverage
    for region, words in region_keywords.items():
        regional_coverage = sum(1 for keyword in words if keyword in combined_text) / len(words)
        regional_keyword_coverage[region] = regional_coverage

    total_regional_keywords_found = sum(1 for region in region_keywords if any(keyword in combined_text for keyword in region_keywords[region]))
    total_regional_keywords = sum(len(words) for words in region_keywords.values())

    scores["regional_completeness"] = np.mean(list(regional_keyword_coverage.values()))

    return scores

# Streamlit UI
st.title("🎄 Weihnachts-Traditionen: Kontextbewertung für KI")

st.markdown("""
    <style>
    .description-section {
        padding: 20px;
        border-radius: 5px;
    }
    </style>
""", unsafe_allow_html=True)

st.markdown(
    """
    <div class="description-section">
    <h2>📝 Aufgabe und Hintergrund</h2>
    <p>In dieser Übung lernst du, wie du Informationen effektiv auswählst und auswertest, um eine abgerundete Antwort für ein KI-Modell zu erstellen.
    Diese Aufgabe hilft dir, die wichtigsten Informationsquellen zu einem Thema zu finden, und vertieft dein Verständnis dafür, wie du einer KI einen Kontext geben kannst, um bessere und relevantere Antworten zu erhalten.</p>

    <h3>🎄 Der Kontext der Weihnachtsfrage:</h3>
    <p>Weihnachten wird in Deutschland mit vielen Traditionen gefeiert, von denen einige allgemein bekannt und andere regionalspezifisch sind.
    Indem du recherchierst und relevante Inhalte auswählst, hilfst du der KI, eine Antwort zu formulieren, die den Charakter und die Vielfalt von Weihnachten in Deutschland einfängt.</p>

    <h3>🎯 Ziel der Aufgabe:</h3>
    <p>Die Aufgabe besteht darin, verschiedene Suchstrategien auszuprobieren und Links zu Wikipedia-Artikeln zu sammeln, die die folgende Frage optimal beantworten:</p>
    <blockquote><strong>„Welche unterschiedlichen Bräuche, Symbole und Traditionen prägen das Weihnachtsfest in Deutschland, und welche regionalen Unterschiede gibt es dabei?“</strong></blockquote>

    <h3>🔍 Schritt-für-Schritt-Anleitung:</h3>
    <ol>
      <li>Beginne mit einer einfachen Suchanfrage auf Wikipedia (z.B. site:de.wikipedia.org Weihnachten).</li>
      <li>Um spezifischere Artikel zu finden, nutze präzise Suchtechniken wie:
        <ul>
          <li><strong>Schlüsselwörter:</strong> Mach die Suchanfrage spezifischer oder allgemeiner.</li>
          <li><strong>Exakte Übereinstimmungen:</strong> Nutze Anführungszeichen für exakte Phrasen (z.B. „Weihnachtsbräuche in Norddeutschand“).</li>
          <li><strong>Textsuche:</strong> Suche nach Artikeln, die den Begriff im Text enthalten (z.B. intext: Weihnachten).</li>
          <li><strong>Artikelanzahl begrenzen:</strong> Konzentriere dich auf die besten Ergebnisse.</li>
        </ul>
      </li>
    </ol>

    <h3>📋 Kriterien für die Bewertung:</h3>
    <p>Nachdem du bis zu 8 Wikipedia-Links eingegeben hast, wird die Anwendung die Artikel anhand dieser Kriterien bewerten:</p>
    <ul>
      <li><strong>Relevanz:</strong> Passt der Artikelinhalt gut zur Frage?</li>
      <li><strong>Vollständigkeit:</strong> Werden alle wichtigen Begriffe erwähnt?</li>
      <li><strong>Regionale Abdeckung:</strong> Werden regionale Unterschiede beschrieben?</li>
    </ul>

    <h3>🎓 Warum ist dies wichtig?</h3>
    <p>Diese Aufgabe veranschaulicht, wie du eine künstliche Intelligenz unterstützen kannst, indem du sorgfältig ausgewählte Informationen bereitstellst, um eine detaillierte Antwort zu erstellen.
    Du erfährst auch, wie Weihnachten in Deutschland gefeiert wird und bekommst einen Einblick in die einzigartigen regionalen Bräuche.
    Die Auswahl der passenden Quellen ist wichtig, um sich ein vollständiges und genaues Bild von einem Thema zu machen.</p>
    </div>
    """,
    unsafe_allow_html=True
)

st.write("---")
st.subheader("🔗 Gib die Artikel-Links ein")

# Input for Wikipedia links
num_links = st.number_input("Wie viele Artikel willst du eingeben? (1-8)", min_value=1, max_value=8, value=1)
urls = []
for i in range(num_links):
    url = st.text_input(f"Link zu Artikel {i + 1}", placeholder="Wikipedia URL")
    urls.append(url)

if st.button("Bewertung durchführen"):
    student_texts = []

    # Fetch and process content from each URL
    for url in urls:
        if url:
            content = fetch_wikipedia_content(url)
            if content:
                student_texts.append(content)

    if len(student_texts) >= 1:
        scores = evaluate_contexts(student_texts)

         # Display scores and explanations
        st.write("---")
        st.subheader("📊 Bewertungsergebnisse")
        st.write(f"**Relevanz:** {scores['relevance']:.2f} - Übereinstimmung der Inhalte mit der Fragestellung.")
        st.write(f"**Vollständigkeit:** {scores['completeness']:.2f} - Abdeckung aller wichtigen Schlüsselbegriffe.")
        st.write(f"**Regionale Abdeckung:** {scores['regional_completeness']:.2f} - Traditionen aus verschiedenen Regionen abgedeckt.")

        # Detailed feedback
        feedback = []
        if scores["relevance"] > 0.8:
          feedback.append("Hohe Relevanz der gefundenen Artikel")
        if scores["relevance"] > 0.6 and scores["relevance"] <= 0.8:
          feedback.append("Die Relevanz der gefundenen Artikel ist bereits gut aber könnte noch verbessert werden")
        if scores["relevance"] <=0.6:
          feedback.append("Die Relevanz der gefundenen Artikel ist nicht hoch und muss verbessert werden.")
        if scores["completeness"] > 0.8:
          feedback.append("Hohe Vollständigkeit der gefundenen Artikel")
        if scores["completeness"] > 0.5 and scores["completeness"] <= 0.8:
          feedback.append("Die Vollständigkeit ist bereits hoch aber einige wichtige Begriffe fehlen noch.")
        if scores["completeness"] <= 0.5:
          feedback.append("Die Vollständigkeit gering; wichtige Begriffe fehlen.")
        if scores["regional_completeness"] > 0.7:
          feedback.append("Regionale Unterschiede sind ausreichend abgedeckt.")
        if scores["regional_completeness"] > 0.5 and scores["regional_completeness"] <= 0.7:
          feedback.append("Regionale Unterschiede sind bereits gut abgedeckt aber könntent noch verbessert werden.")
        if scores["regional_completeness"] <= 0.5:
          feedback.append("Regionale Unterschiede sind nicht ausreichend abgedeckt.")

        if feedback:
            st.write("---")
            st.subheader("🔍 Feedback")
            for item in feedback:
                st.write("- " + item)
        if scores["relevance"] > 0.8 and scores["completeness"] > 0.8 and scores["regional_completeness"] > 0.7:
            st.write("🎉 Die Auswahl der Artikel ist sehr gut!")
    else:
        st.error("Bitte gib einen Link zu Wikipedia-Artikeln ein.")