Update app.py

app.py

@@ -60,7 +60,7 @@ model_dict = {
 
 pipeline_model = load_pipeline(model_dict[task], task)
 
-def process_entities(entities):
+def process_entities(entities, text):
     """
     Varlıkları birleştirip anlamlı bir şekilde düzenler.
     """
@@ -68,7 +68,7 @@ def process_entities(entities):
     current_entity = None
     
     for entity in entities:
-        if 'entity' in entity and entity['entity'].startswith('I-'):
+        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']
@@ -104,22 +104,21 @@ if st.button("Çalıştır") and input_text:
         # Process entities
         processed_entities = []
         for entity in output:
-            if 'entity' in entity:
-                word = entity.get('word', '')
-                label = entity.get('entity', '')
-                score = entity.get('score', 0.0)
-                start = entity.get('start', 0)
-                end = entity.get('end', 0)
-                processed_entities.append({
-                    'word': word,
-                    'label': label,
-                    'score': score,
-                    'start': start,
-                    'end': end
-                })
+            word = entity['word']
+            label = entity['entity']
+            score = entity['score']
+            start = entity['start']
+            end = entity['end']
+            processed_entities.append({
+                'word': word,
+                'label': label,
+                'score': score,
+                'start': start,
+                'end': end
+            })
         
         # Aggregate entities
-        df = pd.DataFrame(process_entities(processed_entities))
+        df = pd.DataFrame(process_entities(processed_entities, input_text))
         st.subheader("Tanımlanan Varlıklar")
         st.dataframe(df)
         
@@ -129,4 +128,26 @@ if st.button("Çalıştır") and input_text:
             last_end = 0
             for item in text_data:
                 if item['start'] > last_end:
-                    formatted_text += original_text[
+                    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'):
+                    color = 'green'
+                else:
+                    color = 'gray'
+                formatted_text += f"<span style='color:{color}; font-weight: bold;'>{word} ({label}, {score:.2f})</span>"
+                last_end = item['end']
+            if last_end < len(original_text):
+                formatted_text += original_text[last_end:]
+            return formatted_text
+
+        formatted_text = format_text(process_entities(processed_entities, input_text), input_text)
+        st.subheader("Analiz Edilen Metin")
+        st.markdown(f"<p>{formatted_text}</p>", unsafe_allow_html=True)
+    elif task == "Metin Oluşturma":
+        output = pipeline_model(input_text, max_length=100, num_return_sequences=1)
+        st.subheader("Oluşturulan Metin")
+        st.write(output[0]['generated_text'])