Update app.py

app.py

@@ -1,27 +1,60 @@
+import gradio as gr
+
 # Positive Wörter laden
 positive = {}
 with open("SentiWS_v2.0_Positive.txt", encoding="utf-8") as f:
     for line in f:
-        wort, score = line.strip().split("\t")
-        positive[wort.lower()] = float(score)
+        parts = line.strip().split("\t")
+        if len(parts) >= 2:
+            wort = parts[0].split("\\")[0].lower()
+            score = float(parts[1])
+            positive[wort] = score
+            # Flexionsformen hinzufügen
+            if len(parts) > 2:
+                for form in parts[2].split(","):
+                    positive[form.lower()] = score
 
 # Negative Wörter laden
 negative = {}
 with open("SentiWS_v2.0_Negative.txt", encoding="utf-8") as f:
     for line in f:
-        wort, score = line.strip().split("\t")
-        negative[wort.lower()] = float(score)
+        parts = line.strip().split("\t")
+        if len(parts) >= 2:
+            wort = parts[0].split("\\")[0].lower()
+            score = float(parts[1])
+            negative[wort] = score
+            if len(parts) > 2:
+                for form in parts[2].split(","):
+                    negative[form.lower()] = score
+# Analysefunktion
+def analyze_sentiment(text):
+    words = text.lower().split()
+    pos_score = sum(positive.get(w, 0) for w in words)
+    neg_score = sum(negative.get(w, 0) for w in words)
 
-import gradio as gr
-# Interface erstellen
-gr.Interface(
-    fn=lambda text: (
-        "✅ Positiv" if sum(positive.get(w, 0) for w in text.lower().split()) > abs(sum(negative.get(w, 0) for w in text.lower().split()))
-        else "❌ Negativ" if sum(negative.get(w, 0) for w in text.lower().split()) > abs(sum(positive.get(w, 0) for w in text.lower().split()))
-        else "➖ Neutral"
-    ),
-    inputs="text",
-    outputs="text",
-    title="Textklassifikation mit SentiWS",
-    description="Gib einen Satz ein und erfahre, ob er positiv oder negativ klingt."
-).launch()
+    if pos_score > abs(neg_score):
+        label = "✅ Positiv"
+    elif neg_score > abs(pos_score):
+        label = "❌ Negativ"
+    else:
+        label = "➖ Neutral"
+
+    return {
+        "Ergebnis": label,
+        "Positiv-Score": round(pos_score, 4),
+        "Negativ-Score": round(neg_score, 4)
+    }
+
+# Gradio-Interface
+with gr.Blocks() as demo:
+    gr.Markdown("# Textklassifikation mit SentiWS")
+    gr.Markdown("Gib einen Satz ein und erfahre, ob er positiv oder negativ klingt.")
+
+    with gr.Row():
+        inp = gr.Textbox(label="Dein Text", placeholder="Schreibe hier...")
+        out = gr.Label(label="Analyse")
+
+    btn = gr.Button("Analysieren")
+    btn.click(analyze_sentiment, inputs=inp, outputs=out)
+
+demo.launch()