Update app.py

app.py

@@ -60,6 +60,38 @@ model_dict = {
 
 pipeline_model = load_pipeline(model_dict[task], task)
 
+def process_entities(entities, text):
+    """
+    Varlıkları birleştirip anlamlı bir şekilde düzenler.
+    """
+    processed_entities = []
+    current_entity = None
+    
+    for entity in entities:
+        if entity['entity'].startswith('I-'):
+            if current_entity and current_entity['label'] == entity['entity']:
+                current_entity['word'] += entity['word'].replace('##', '')
+                current_entity['end'] = entity['end']
+                current_entity['score'] = max(current_entity['score'], entity['score'])
+            else:
+                if current_entity:
+                    processed_entities.append(current_entity)
+                current_entity = {
+                    'label': entity['entity'],
+                    'word': entity['word'].replace('##', ''),
+                    'start': entity['start'],
+                    'end': entity['end'],
+                    'score': entity['score']
+                }
+        else:
+            if current_entity:
+                processed_entities.append(current_entity)
+                current_entity = None
+    if current_entity:
+        processed_entities.append(current_entity)
+    
+    return processed_entities
+
 if st.button("Çalıştır") and input_text:
     if task in ["Metin Sınıflandırma", "Duygu Analizi"]:
         output = pipeline_model(input_text)
@@ -69,42 +101,9 @@ if st.button("Çalıştır") and input_text:
     elif task == "Metin Analizi":
         output = pipeline_model(input_text)
         
-        # Varlıkları birleştirmek için işlev
-        def process_entities(entities, text):
-            result = []
-            current_entity = None
-            for entity in entities:
-                if entity['entity'].startswith('I-'):
-                    if current_entity and current_entity['Label'] == entity['entity']:
-                        current_entity['Word'] += entity['word']
-                        current_entity['End'] = entity['end']
-                        current_entity['Score'] = max(current_entity['Score'], entity['score'])
-                    else:
-                        if current_entity:
-                            result.append(current_entity)
-                        current_entity = {
-                            'Label': entity['entity'],
-                            'Score': entity['score'],
-                            'Word': entity['word'],
-                            'Start': entity['start'],
-                            'End': entity['end']
-                        }
-                else:
-                    if current_entity:
-                        result.append(current_entity)
-                    current_entity = {
-                        'Label': entity['entity'],
-                        'Score': entity['score'],
-                        'Word': entity['word'],
-                        'Start': entity['start'],
-                        'End': entity['end']
-                    }
-            if current_entity:
-                result.append(current_entity)
-            return result
-
-        entities = process_entities(output, input_text)
-        df = pd.DataFrame(entities)
+        # Process entities
+        processed_entities = process_entities(output, input_text)
+        df = pd.DataFrame(processed_entities)
         st.subheader("Tanımlanan Varlıklar")
         st.dataframe(df)
         
@@ -113,11 +112,11 @@ if st.button("Çalıştır") and input_text:
             formatted_text = ""
             last_end = 0
             for item in text_data:
-                if item['Start'] > last_end:
-                    formatted_text += original_text[last_end:item['Start']]
-                word = item['Word']
-                label = item['Label']
-                score = item['Score']
+                if item['start'] > last_end:
+                    formatted_text += original_text[last_end:item['start']]
+                word = item['word']
+                label = item['label']
+                score = item['score']
                 if label.startswith('I-PER'):
                     color = 'blue'
                 elif label.startswith('I-MISC'):
@@ -125,12 +124,12 @@ if st.button("Çalıştır") and input_text:
                 else:
                     color = 'gray'
                 formatted_text += f"<span style='color:{color}; font-weight: bold;'>{word} ({label}, {score:.2f})</span>"
-                last_end = item['End']
+                last_end = item['end']
             if last_end < len(original_text):
                 formatted_text += original_text[last_end:]
             return formatted_text
 
-        formatted_text = format_text(entities, input_text)
+        formatted_text = format_text(processed_entities, input_text)
         st.subheader("Analiz Edilen Metin")
         st.markdown(f"<p>{formatted_text}</p>", unsafe_allow_html=True)
     elif task == "Metin Oluşturma":