import torch
from transformers import pipeline, set_seed
import re

set_seed(42)
model = pipeline("text-generation", model="gpt2")

def generate_script(conflict):
    characters = generate_characters(conflict)
    print(characters)
    plot = generate_plot(conflict, characters)
    print(plot)
    dialogue = generate_dialogue(conflict, characters, plot)
    print(dialogue)
    return characters, plot, dialogue

def generate_characters(conflict):
    characters = model(f"""Characters for a conflict
    conflict: {conflict}
    list of characters:
    1.""", max_length=100, num_return_sequences=1)
    characters_text = characters[0]['generated_text']
    character_list = re.findall(r'\d+\.(.*?)\.', characters_text)
    return character_list

def generate_plot(conflict, characters):
    plot = model("Plot for a conflict: " + conflict, max_length=500, num_return_sequences=1)
    plot_text = plot[0]['generated_text']
    return plot_text.split("\n", 1)[1]

def generate_dialogue(conflict, characters, plot):
    dialogue = model(f"""Dialogue for a conflict:
    conflict = {conflict}
    characters = {characters[0]} and {characters[1]}
    
    {characters[0]}: """, max_length=1000, num_return_sequences=1)
    dialouge_text = dialogue[0]['generated_text']
    return dialouge_text

if __name__ == '__main__':
    conflict = "Two siblings compete for the throne after the death of their father, the king."
    characters, plot, dialogue = generate_script(conflict)
    print("Conflict:", conflict)
    print("\nCharacters:")
    for i, char in enumerate(characters):
        print(f"{i + 1}. {char}")
    print("\nPlot:")
    print(plot)
    print("\nDialogue:")
    print(dialogue)
นำเข้าไฟฉายจากหม้อแปลงนำเข้าไปป์ไลน์ set_seed นำเข้าใหม่ set_seed (42) model = ไปป์ไลน์ ("text-generator", model = "gpt2") def Gene_script (ขัดแย้ง): อักขระ = Generate_Characters (ขัดแย้ง) พิมพ์ (อักขระ) plot = Generate_plot ( ความขัดแย้ง, ตัวอักษร) พิมพ์(พล็อต) บทสนทนา = Generate_dialogue(ขัดแย้ง, ตัวละคร, พล็อต) พิมพ์(บทสนทนา) กลับตัวละคร, พล็อต, บทสนทนา def Generate_Characters(ขัดแย้ง): ตัวละคร = model(f"""อักขระสำหรับความขัดแย้ง ความขัดแย้ง: {ความขัดแย้ง } รายการอักขระ: 1.""", max_length=100, num_return_sequences=1) character_text = character[0]['generated_text'] character_list = re.findall(r'\d+\.(.*?)\.' , character_text) ส่งคืน character_list def create_plot(ความขัดแย้ง, อักขระ): plot = model("Plot for aขัดแย้ง: " + allowance, max_length=500, num_return_sequences=1) plot_text = plot[0]['generated_text'] return plot_text.split ("\n", 1)[1] def Generate_dialogue(ความขัดแย้ง, ตัวละคร, โครงเรื่อง): บทสนทนา = model(f"""บทสนทนาสำหรับความขัดแย้ง: ความขัดแย้ง = {ความขัดแย้ง} อักขระ = {อักขระ[0]} และ {อักขระ [1]} {ตัวละคร[0]}: """, max_length=1000, num_return_sequences=1) dialouge_text = Dialogue[0]['generated_text'] ส่งคืน dialouge_text ถ้า __name__ == '__main__':ความขัดแย้ง = "พี่น้องสองคนแข่งขันกัน เพื่อครองบัลลังก์ภายหลังพระราชบิดาสิ้นพระชนม์" ตัวละคร, โครงเรื่อง, บทสนทนา = Generate_script(ความขัดแย้ง) print("ความขัดแย้ง:", ความขัดแย้ง) print("\nตัวละคร:") สำหรับ i, ถ่านในการแจงนับ (อักขระ): print(f"{i + 1}. {char} ") print("\nPlot:") พิมพ์(plot) print("\nDialogue:") พิมพ์(บทสนทนา)
spam fails_task lang_mismatch pii not_appropriate hate_speech sexual_content quality toxicity humor helpfulness creativity violence