Update app.py

app.py

@@ -69,32 +69,38 @@ if st.button("Çalıştır") and input_text:
     elif task == "Metin Analizi":
         output = pipeline_model(input_text)
         
-        # Varlıkları ayırmak ve birleştirmek
+        # Varlıkları birleştirmek için işlev
         def process_entities(entities, text):
-            # Varlıkları birleştirmek için bir yapı oluştur
             result = []
+            current_entity = None
             for entity in entities:
                 if entity['entity'].startswith('I-'):
-                    if result and result[-1]['Label'] == entity['entity']:
-                        result[-1]['Word'] += entity['word']
-                        result[-1]['End'] = entity['end']
-                        result[-1]['Score'] = max(result[-1]['Score'], entity['score'])
+                    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:
-                        result.append({
+                        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:
-                    result.append({
+                    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)
@@ -118,4 +124,16 @@ if st.button("Çalıştır") and input_text:
                     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(entities, 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'])