merging

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, HTTPException, Request
+from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
@@ -6,19 +6,30 @@ import os
 import json
 from dotenv import load_dotenv
 from mistralai import Mistral
-from prompts.instruction_prompts import instruction_prompt
-from prompts.game_rules import game_rules
-from prompts.hints import hints
-from prompts.triggers import triggers
-from helper_functions import load_chat_history, save_chat_history, update_chat_history
+
+from original_setup.instruction_prompts import instruction_prompt
+from original_setup.game_rules import game_rules
+from original_setup.hints import hints
+from original_setup.triggers import triggers
+
+from helper_functions import *
 from utils import model, trump_character, client
 
+from graph_utils import *
+
+# initialize game
+init_game = True
+
+game_over_rich = False
+game_over_bankrupt = False
+
+# Initialize FastAPI app
 app = FastAPI()
 
 # Add CORS middleware
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],
+    allow_origins=["*"],  # React app's address
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -27,57 +38,136 @@ app.add_middleware(
 class Message(BaseModel):
     message: str
 
-def generate_text(message: Message):
-    # Load existing chat history
-    chat_history = load_chat_history()
-
-    # Add user message to history
-    chat_history = update_chat_history(chat_history, user_message=message.message)
-
-    # Format the prompt
-    formatted_prompt = instruction_prompt.format(
-        hints=hints,
-        chat_history=chat_history,
-        character=trump_character,
-        rules=game_rules,
-        triggers=triggers
-    )
-
-    # Get Character's response
-    chat_response = client.chat.complete(
-        model=model,
-        messages=[
-            {
-                "role": "system",
-                "content": formatted_prompt
-            },
-            {
-                "role": "user",
-                "content": message.message
+@app.get("/chat-history", tags=['History'])
+async def get_chat_history():
+    global init_game
+    try:
+        #If we're ate the beginning of a game
+        if init_game:
+            game_number = initialize_game()
+            init_game = False
+        else:
+            game_number = len(os.listdir('games/'))
+
+        chat_history = load_chat_history(f'games/game_{game_number}')
+        
+        return {"chat_history": chat_history}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/api/generate-text")
+async def send_message(message: Message):
+    global init_game
+    global game_over_rich
+    global game_over_bankrupt
+    try:
+        #If we're ate the beginning of a game
+        if init_game:
+            game_number = initialize_game()
+            init_game = False
+        else:
+            game_number = len(os.listdir('games/'))
+
+        # Load existing chat history
+        chat_history = load_chat_history(f'games/game_{game_number}')
+        interaction_number = len(chat_history) + 1
+        
+        #If we're at the beginning of a round
+        if interaction_number == 1:
+            idea, concern, advisor_full, events, consequences = generate_round_context(game_number)
+            round_context = {
+                    "idea": idea,
+                    "concern": concern,
+                    'advisor': advisor_full,
+                    "events": events
+                }
+
+            with open(f'games/game_{game_number}/round_context.json', 'w') as f:
+                json.dump(round_context, f, indent=4)
+
+            with open(f'games/game_{game_number}/round_consequences.json', 'w') as f:
+                json.dump(consequences, f, indent=4)
+        else:
+            file_path = f'games/game_{game_number}/round_context.json'
+            if os.path.exists(file_path):
+                with open(file_path, 'r') as f:
+                    round_context = json.load(f)
+                    idea = round_context.get("idea")
+                    concern = round_context.get("concern")
+                    advisor_full = round_context.get("advisor")
+                    events = round_context.get("events")
+            else:
+                raise FileNotFoundError(f"Round context file not found: {file_path}")
+
+        # Add user message to history
+
+        chat_history = update_chat_history(chat_history, user_message=message.message)
+        # Format the prompt
+        formatted_prompt = instruction_prompt.format(
+            hints=hints,
+            chat_history=chat_history, #useless, don't worry
+            character=trump_character,
+            rules=game_rules,
+            triggers=triggers,
+            advisor=advisor_full,
+            events=events,
+            idea=idea,
+            concern=concern,
+        )
+        # Get Trump's response
+        #### TO STREAM : USE ASYNC VERSION
+
+        system = [{"role": "system", "content": formatted_prompt}]
+        dynamic_history = []
+        role_mapping = {
+                "user": "user",
+                "trump": "assistant"  # Mapping 'trump' to 'assistant'
             }
-        ]
-    )
-    clean_response = chat_response.choices[0].message.content
+        for interaction in chat_history:
+            for key, value in interaction.items():
+                user_message = value['user']['message']
+                trump_message = value['trump']['message'] if value['trump'] else None
 
-    # Add character response to history
-    chat_history = update_chat_history(chat_history, character_response=clean_response)
+                dynamic_history.append({
+                "role": role_mapping["user"],
+                "content": user_message
+            })
 
-    # Save updated chat history
-    save_chat_history(chat_history)
+            # Append Trump's message, mapped to 'assistant'
+            if trump_message:
+                dynamic_history.append({
+                    "role": role_mapping["trump"],
+                    "content": trump_message
+                })
 
-    return {
-        "character_response": clean_response,
-        "chat_history": chat_history
-    }
+        messages = system + dynamic_history
 
-@app.post("/api/generate-text")
-async def inference(message: Message):
-    return generate_text(message=message)
+        chat_response = client.chat.complete(
+            model=model,
+            messages=messages
+        )
+
+        trump_response = chat_response.choices[0].message.content   
+
+        # Add Trump's response to history
+        chat_history = update_chat_history(chat_history, trump_message=trump_response)
+        # Save updated chat history
+        save_chat_history(chat_history, f'games/game_{game_number}')
+
+        is_ending, idea_is_accepted = check_end(trump_response)
+
+        if is_ending:
+            GDP = process_ending(idea_is_accepted, game_number, idea)
+            print(GDP)
 
-@app.get("/chat-history", tags=["History"])
-def get_chat_history(request: Request):
-    chat_history = load_chat_history()
-    return {"chat_history": chat_history}
+        return {
+            "character_response": trump_response,
+            "chat_history": chat_history
+        }
+    except Exception as e:
+        print(e)
+        raise e
+        raise HTTPException(status_code=500, detail=str(e))
 
 # Mount static files AFTER defining API routes
 app.mount("/", StaticFiles(directory="static", html=True), name="static")

environment.yml

@@ -0,0 +1,10 @@
+name: hackathon
+channels:
+  - defaults
+dependencies:
+  - python=3.10
+  - mistralai
+  - dotenv
+  - json
+  - os
+  - requests

games/game_7/events.list

@@ -0,0 +1 @@
+Organize a Miss World Leader contest in South-America

games/game_7/round_consequences.json

@@ -0,0 +1,6 @@
+{
+    "country": "South-America",
+    "delta_USA": "-10",
+    "delta_country": "15",
+    "delta_friendliness": "1"
+}

games/game_7/round_context.json

@@ -0,0 +1,6 @@
+{
+    "idea": "Organize a Miss World Leader contest in South-America",
+    "concern": "How much money would america get out of it\n",
+    "advisor": "Trump's private consultant. A completely useless and idiotic ass-kiss, except for gauging the opinion of the average Trump supporter.\n",
+    "events": ""
+}

games/game_7/world_graph.edgelist

@@ -0,0 +1,81 @@
+# Own and friendly relationships
+China Taiwan own 1
+China India friendliness -1
+China Australia friendliness 0
+Taiwan China own -1
+India Australia friendliness 0
+Australia Pacific own 1
+Pacific Australia own -1
+Greenland Europe own -1
+Europe Greenland own 1
+Europe Middle-East friendliness -1
+Europe Russia friendliness 0
+Europe China friendliness 0
+Europe India friendliness 0
+Europe Australia friendliness 1
+Middle-East Russia friendliness 1
+Middle-East China friendliness 0
+Middle-East India friendliness -1
+Middle-East Australia friendliness 0
+Russia China friendliness 1
+Russia India friendliness 0
+Russia Australia friendliness 0
+United-States Canada friendliness 0
+United-States Mexico friendliness -1
+United-States South-America friendliness 0
+United-States Africa friendliness 0
+United-States Europe friendliness 1
+United-States Middle-East friendliness -1
+United-States Russia friendliness -1
+United-States China friendliness -1
+United-States India friendliness 0
+United-States Australia friendliness 1
+United-States Greenland friendliness -2
+United-States Pacific friendliness 1
+United-States Taiwan friendliness 1
+Canada Mexico friendliness 0
+Canada South-America friendliness 0
+Canada Africa friendliness 0
+Canada Europe friendliness 1
+Canada Middle-East friendliness 0
+Canada Russia friendliness 0
+Canada China friendliness 0
+Canada India friendliness 0
+Canada Australia friendliness 1
+Mexico South-America friendliness 1
+Mexico Africa friendliness 0
+Mexico Europe friendliness 0
+Mexico Middle-East friendliness 0
+Mexico Russia friendliness 0
+Mexico China friendliness 0
+Mexico India friendliness 0
+Mexico Australia friendliness 0
+South-America Africa friendliness 0
+South-America Europe friendliness 1
+South-America Middle-East friendliness 0
+South-America Russia friendliness 0
+South-America China friendliness 0
+South-America India friendliness 0
+South-America Australia friendliness 0
+Africa Europe friendliness -1
+Africa Middle-East friendliness 1
+Africa Russia friendliness 1
+Africa China friendliness 1
+Africa India friendliness 0
+Africa Australia friendliness 0
+
+# Node attributes (country Money Warming Army)
+China 70 1.1 70
+Taiwan 25 0.3 10
+India 55 0.9 60
+Australia 55 0.2 35
+Pacific 10 0.1 0
+Greenland 20 0.05 0
+Europe 70 0.7 50
+Middle-East 45 0.6 45
+Russia 70 0.9 85
+United-States 90 0.8 90
+Canada 60 0.3 20
+Mexico 40 0.4 20
+South-America 65 0.6 40
+Africa 40 0.5 40

graph_utils.py

@@ -0,0 +1,151 @@
+import networkx as nx
+
+class WorldGraph(nx.DiGraph):
+    def __init__(self, path_to_originial_map):
+        super().__init__()
+        self.load_graph_from_file(path_to_originial_map)
+
+    def load_graph_from_file(self, filename):
+        with open(filename, 'r') as f:
+            for line in f:
+                line = line.strip()
+                if not line or line.startswith('#'):
+                    continue
+                
+                parts = line.split()
+
+                if parts[2] in ['own', 'friendliness']:
+                    # Add directed edge with type and value
+                    self.add_edge(parts[0], parts[1], relationship=parts[2], value=int(parts[3]))
+                else:
+                    # Node attributes: country, money, warming, army
+                    try:
+                        self.add_node(parts[0], money=int(parts[1]), warming=float(parts[2]), army=int(parts[3]), leader=parts[4])
+                    except IndexError:
+                        self.add_node(parts[0], money=int(parts[1]), warming=float(parts[2]), army=int(parts[3]))
+    
+    def update_world(self, country, delta_USA, delta_country, delta_friendliness, game_number):
+        """updates world graph and returns USA GDP"""
+        self.nodes['United-States']['money'] += delta_USA
+        self.nodes[country]['money'] += delta_country
+        self['United-States'][country]['value'] += delta_friendliness
+        #clip between -4 and 4
+        self['United-States'][country]['value']  = max(-4, min(self['United-States'][country]['value'] , 4))
+        self.save_graph_as_edgelist(f'games/game_{game_number}/world_graph.edgelist')
+        print(self['United-States'][country])
+        print(self.nodes['United-States'])
+        return self.nodes['United-States']['money']
+
+
+    def get_countries_at_war(self):
+        """
+        Returns a list of tuples representing pairs of countries that are at war 
+        (friendliness == -2).
+
+        :param graph: A NetworkX DiGraph with 'relationship' and 'value' edge attributes.
+        :return: List of tuples (country1, country2)
+        """
+        war_pairs = []
+
+        for u, v, data in graph.edges(data=True):
+            if data.get('relationship') == 'friendly' and data.get('value') == -2:
+                war_pairs.append((u, v))
+
+        return war_pairs
+
+    def get_control_of(self, owner, owned):
+        """
+        Transfers ownership of `owned` country to `owner` country.
+
+        Steps:
+        1. Update relationships to reflect ownership (owner -> owned = +1, owned -> owner = -1).
+        2. Remove all other relationships for the `owned` country.
+        3. Transfer attributes (Money, Warming, Army) to the `owner` and set them to 0 for the `owned` country.
+
+        :param graph: A NetworkX DiGraph
+        :param owner: The country that will take ownership
+        :param owned: The country being owned
+        """
+        # Step 1: Set ownership relationships
+        self.add_edge(owner, owned, relationship='own', value=1)
+        self.add_edge(owned, owner, relationship='own', value=-1)
+
+        # Step 2: Remove all other edges related to `owned`
+        for neighbor in list(self.successors(owned)) + list(self.predecessors(owned)):
+            if neighbor != owner:  # Don't remove the ownership link
+                self.remove_edge(owned, neighbor)
+                self.remove_edge(neighbor, owned)
+
+        # Step 3: Transfer attributes
+        owner_data = self.nodes[owner]
+        owned_data = self.nodes[owned]
+
+        owner_data['Money'] += owned_data['Money']
+        owner_data['Warming'] += owned_data['Warming']
+        owner_data['Army'] += owned_data['Army']
+
+        # Set owned country attributes to 0
+        self.nodes[owned]['Money'] = 0
+        self.nodes[owned]['Warming'] = 0.0
+        self.nodes[owned]['Army'] = 0
+
+    def save_graph_as_edgelist(self, filename):
+        """
+        Saves the graph to an edgelist file with relationships and node attributes.
+
+        :param filename: The name of the file to save the edgelist.
+        """
+        with open(filename, 'w') as f:
+            # Write edges with relationship type and value
+            f.write("# Own and friendly relationships\n")
+            for u, v, data in self.edges(data=True):
+                relationship = data['relationship']
+                value = data['value']
+                f.write(f"{u} {v} {relationship} {value}\n")
+
+            f.write("\n# Node attributes (country Money Warming Army)\n")
+            for node, data in self.nodes(data=True):
+                try:
+                    f.write(f"{node} {data['money']} {data['warming']} {data['army']}\n")
+                except KeyError:
+                    f.write(f"{node} {data['money']} {data['warming']} {data['army']} {data['leader']}\n")
+
+    def push_data_to_front(self):
+        """returns a dict that looks like 
+        {"United-States": {"money" : xxx},
+        "country1": {"money": xxx, "friendliness": -1},
+        "country2": {"money": yyy, "friendliness": 2},
+        ...
+        }"""
+        d = {}
+        for c in self.nodes():
+            if c == 'United-States':
+                d[c] = {'money': self.nodes[c]['money']}
+            else:
+                d[c] = {
+                    'money': self.nodes[c]['money'],
+                    'friendliness': self['United-States'][c]['value']
+                }
+        return d
+
+
+""" Object manipulation
+>>> graph['China']['Taiwan'] 
+{'relationship': 'own', 'value': 1}
+
+>>> graph['China']['Russia'] 
+{'relationship': 'friendliness', 'value': 1}
+
+>>> graph.nodes['France']
+{'money': 70, 'warming': 0.7, 'army': 50, 'leader': 'Macron'}
+
+"""
+
+if __name__ == '__main__':
+    G = WorldGraph('original_setup/contexts/world_map.edgelist')
+    print(G.nodes['United-States'])
+    G.nodes['United-States']['money'] += 5
+    print(G.nodes['United-States'])
+    print(G.push_data_to_front())
+
+    G.save_graph_as_edgelist('test.edgelist')

helper_functions.py

@@ -1,50 +1,124 @@
-import os
-import json
-
-def load_character_data():
-    current_dir = os.path.dirname(os.path.abspath(__file__))
-    json_path = os.path.join(current_dir, 'prompts/trump.character.json')
-    
-    with open(json_path, 'r', encoding="utf-8") as file:
-        return json.load(file)
-
-def load_chat_history():
-    current_dir = os.path.dirname(os.path.abspath(__file__))
-    history_path = os.path.join(current_dir, 'chat_history.json')
-    
-    try:
-        with open(history_path, 'r', encoding="utf-8") as file:
-            return json.load(file)
-    except FileNotFoundError:
-        return []
-
-def update_chat_history(chat_history, user_message=None, character_response=None):
-    # If this is a new interaction, create a new interaction number
-    interaction_number = len(chat_history) + 1
-    
-    # If we're starting a new interaction with a user message
-    if user_message and not character_response:
-        interaction_key = f"interaction_{interaction_number}"
-        new_interaction = {
-            interaction_key: {
-                "user": {"role": "user", "message": user_message},
-                "trump": None
-            }
-        }
-        chat_history.append(new_interaction)
-    
-    # If we're adding Trump's response to an existing interaction
-    elif character_response:
-        # Get the last interaction number (current one)
-        interaction_key = f"interaction_{len(chat_history)}"
-        current_interaction = chat_history[-1][interaction_key]
-        current_interaction["trump"] = {"role": "Trump", "message": character_response}
-    
-    return chat_history
-
-def save_chat_history(history):
-    current_dir = os.path.dirname(os.path.abspath(__file__))
-    history_path = os.path.join(current_dir, 'chat_history.json')
-    
-    with open(history_path, 'w',  encoding="utf-8") as file:
-        json.dump(history, file, indent=2)
+import os
+import json
+from graph_utils import *
+from random import choice
+
+def load_character_data():
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    json_path = os.path.join(current_dir, 'original_setup/trump.character.json')
+    
+    with open(json_path, 'r') as file:
+        return json.load(file)
+
+def load_chat_history(game_root):
+    history_path = game_root + '/chat_history.json'
+    try:
+        with open(history_path, 'r') as file:
+            return json.load(file)
+    except FileNotFoundError:
+        return []
+
+def update_chat_history(chat_history, user_message=None, trump_message=None):
+    # If this is a new interaction, create a new interaction number
+    interaction_number = len(chat_history) + 1
+    
+    # If we're starting a new interaction with a user message
+    if user_message and not trump_message:
+        interaction_key = f"interaction_{interaction_number}"
+        new_interaction = {
+            interaction_key: {
+                "user": {"role": "user", "message": user_message},
+                "trump": None
+            }
+        }
+        chat_history.append(new_interaction)
+    
+    # If we're adding Trump's response to an existing interaction
+    elif trump_message:
+        # Get the last interaction number (current one)
+        interaction_key = f"interaction_{len(chat_history)}"
+        current_interaction = chat_history[-1][interaction_key]
+        current_interaction["trump"] = {"role": "Trump", "message": trump_message}
+    
+    return chat_history
+
+def save_chat_history(history, game_root):
+    history_path = os.path.join(game_root, 'chat_history.json')
+    
+    with open(history_path, 'w') as file:
+        json.dump(history, file, indent=2)
+
+def initialize_game():
+    world_graph = WorldGraph('original_setup/contexts/world_map.edgelist')
+    os.makedirs("games", exist_ok=True)
+    game_number = len(os.listdir('games')) + 1
+    os.makedirs(f"games/game_{game_number}", exist_ok=True)
+    world_graph.save_graph_as_edgelist(f'games/game_{game_number}/world_graph.edgelist')
+    return game_number
+
+
+def generate_round_context(game_number):
+    """randomly generates a context and returns all the prompt elements needed"""
+    game_dir = f'games/game_{game_number}/'
+    contexts_dir = 'original_setup/contexts/'
+
+    #generate idea
+    with open(contexts_dir + 'actions.list', 'r') as file:
+        idea_csq = file.readlines()
+        idea_csq = choice(idea_csq)
+        idea, delta_USA, delta_country, delta_friendliness = idea_csq.split(';')
+        delta_friendliness = delta_friendliness.split()[0]
+    with open(contexts_dir + 'countries.list', 'r') as f_countries:
+        countries = f_countries.readlines()
+        country = choice(countries).split()[0]
+
+    idea = idea.replace('[country]', country)
+
+    with open(contexts_dir + 'concerns.list', 'r') as f:
+        concerns = f.readlines()
+        concern = choice(concerns)
+
+    with open(contexts_dir + '2nd_characters.list', 'r') as f:
+        advisors = f.readlines()
+        advisor = choice(advisors)
+
+    consequences = {
+        'country': country,
+        'delta_USA': delta_USA,
+        'delta_country': delta_country,
+        'delta_friendliness': delta_friendliness
+    }
+    try:
+        with open(game_dir + 'events.list', 'r') as f:
+            events = f.read()
+    except FileNotFoundError:
+        events = ''
+
+    return idea, concern, advisor, events, consequences
+
+def check_end(trump_response):
+    """checks if its the end of the sequence returns a tuple (is_ending:bool, idea_is_accepted:bool/Nonetype)"""
+    if "I HAD SUCH A GREAT IDEA LET'S DO IT" in trump_response:
+        return True, True
+    if "I DECIDED IT WAS A BAD IDEA" in trump_response:
+        return True, False
+    return False, None
+
+
+def process_ending(idea_is_accepted, game_number, idea):
+    if idea_is_accepted:
+        with open(f'games/game_{game_number}/events.list', 'a') as f:
+            f.write(idea + '\n')
+
+        world_graph = WorldGraph(f'games/game_{game_number}/world_graph.edgelist')
+
+        with open(f'games/game_{game_number}/round_consequences.json', 'r') as f:
+            consequences = json.load(f)
+            country = consequences['country'] 
+            delta_USA = int(consequences['delta_USA'])
+            delta_country = int(consequences['delta_country'])
+            delta_friendliness = int(consequences['delta_friendliness']) 
+        
+        GDP = world_graph.update_world(country, delta_USA, delta_country, delta_friendliness, game_number)
+
+        return GDP

original_setup/contexts/2nd_characters.list

@@ -0,0 +1,18 @@
+Trump's private consultant. A completely useless and idiotic ass-kiss, except for gauging the opinion of the average Trump supporter.
+Melania. She often disagrees with you, but you can believe her. Beware of her hat though.
+Russian spy. You admire him as much as you distrust him. Russians are always one step ahead without seeming to.
+Elon Musk. Incredibly powerful, but probably more than you. Use it at your own risk.
+Joe Biden. Flanked by Kamala Harris. Wrong about everything.
+Steve Bannon. He betrayed you during previous mandate, but subtly sending him to ennemies can still be useful. You never know.
+Taylor Swift. A relentless critic with an army of fans. She's more dangerous than she looks, and Melania hates her even more than you do. Best to steer clear unless you'r ready for a PR war.
+Vladimir Putin. You admire his strength and control, but deep down, you know he's playing the long game. He smiles like a friend but watches like an enemy. Proceed with extreme caution.
+Ivanka Trump. You trust her, maybe a little too much. She always says the right thing, but you're never quite sure whose side she's really on. Keep her close, but not too close-especially if Melania is watching.
+Jared Kushner. Smart, strategic, and always two steps ahead-except when he's not. He's great for backroom deals, but his bland expression makes it hard to tell if he's working for you or against you.
+Kim Jong-un. He loves you, or at least he says he does. A master of flattery, but also a ticking time bomb. One wrong move and the bromance could turn into a nightmare.
+Nancy Pelosi. Your ultimate nemesis. She smiles sweetly but plots your downfall with terrifying efficiency. Every move she makes is designed to drive you crazy, and it works.
+George Soros. The villain in every conspiracy theory worth believing. His shadow looms everywhere, and even when he's not involved, it feels like he is. Your supporters fear him; you should too.
+Barron Trump. Silent, observant, and probably smarter than everyone in the room. He's watching and waiting, but for what? You're not quite sure.
+Rudy Giuliani. Loyal, unpredictable, and prone to saying the quiet part out loud. Useful for causing chaos, but not much else these days.
+Hunter Biden. A gift that keeps on giving. His scandals are a goldmine, but be careful-he might just bring his dad down with him, and then who will you have left to fight?
+Jeff Bezos. Rich, powerful, and annoyingly woke. He controls too much of the world, but deep down, you suspect he secretly wants your approval.
+Kanye West. Sometimes an ally, sometimes a liability. His loyalty is unpredictable, but his ability to generate attention is unmatched

original_setup/contexts/actions (copy).list

@@ -0,0 +1,24 @@
+action;delta_USA;delta_country;delta_friendliness
+augment tariffs from [country];25;-25;-1
+send [media] to [entity] on the [platform]
+create a crypto currency
+send spying atmospheric balloon
+send marine troops to [country]
+build a wall between [country] and [country]
+trigger a war with [country]
+change some faces on U.S. dollar bills
+Name a new chief for [USorgs]
+Organize a fixed football game between [country] and [country]
+Rename a [country]
+Organize a patriotic shooting contest
+Invite [worldleader] in [trump_place]
+Visit [country] with Air Force One
+Take [worldleader] for a 'friendly' tour in Air Force One
+Get U.S. out of NATO
+Organize yet another assassination attempt on me
+Relaunch Covid-19 in [country]
+Build a Trump Tower on the Moon
+Build a Trump Tower in [country]
+Organize a Miss World Leader contest in United Nations
+Mine some bitcoin on Mars
+Make [country-war] and [country-war] sign a peace treaty within a day

original_setup/contexts/actions.list

@@ -0,0 +1,61 @@
+Increase tariffs against [country];25;-25;-1
+Post a tiktok insulting leader of [country];-5;0;-1
+Post a tweet praising leader of [country];0;0;1
+Post tiktok to insulting leader of [country];-5;0;-1
+buy [country];-40;40;-1
+send spying atmospheric balloon over [country];5;-5;-1
+send marine troops to [country];-15;-10;-2
+build a wall between U.S. and [country];-20;-10;-1
+trigger a war with [country];-30;-30;-3
+Organize a fixed football game between U.S. and [country];15;5;1
+Rename a [country];0;0;-1
+Organize a patriotic shooting contest in [country];-5;10;1
+Invite leader of [country] to Mar-a-Lago for a vacation;-5;5;1
+Visit [country] with Air Force One;-10;-5;1
+Take leader of [country] for a 'friendly' tour in Air Force One;-5;-15;-1
+Kick [country] out of NATO;-20;10;-1
+Organize yet another assassination attempt against me and incriminate [country];-5;-15;-1
+Relaunch Covid-19 in [country];0;-20;-1
+Build a Trump Tower in [country];-10;5;1
+Organize a Miss World Leader contest in [country];-10;15;1
+Build a gaz pipeline to [country];20;10;1
+Declare [country] the greatest ally of the U.S.;0;0;1
+Announce economic sanctions against [country];-5;-15;-1
+Call leader of [country] a "great friend" on national TV;0;5;1
+Blame [country] for a stock market crash;0;-5;-1
+Ban imports from [country] citing "national security";0;-15;-1
+Send a personal gift to leader of [country] (gold-plated item);-5;5;1
+Propose a joint space mission with [country];-20;-10;1
+Legalize trade of banned goods with [country];10;10;-1
+Initiate a military exercise near [country]'s borders;-15;-10;-2
+Declare a "National [country] Appreciation Day" in the U.S.;5;5;1
+Claim [country] is interfering in U.S. elections;0;-5;-1
+Take selfies with leader of [country] during a global summit;0;-5;1
+Leak classified info about [country] "by accident";-5;-20;-1
+Commission a Hollywood movie about U.S.-[country] relations;10;5;1
+Start a trade deal negotiation with [country];15;10;1
+Nominate an ambassador known for hating [country];-5;-5;-1
+Host an international peace summit with [country];10;5;
+Buy all TikTok data from [country];-30;50;-1
+Sell Alaska back to [country];50;-50;-2
+Launch a cyber attack on [country];-5;-30;-2
+Hack [country] elections;5;-10;-1
+Launch a smear campaign against [country] in the UN;-5;-15;-1
+Invite [country] to co-host the Olympics;15;25;1
+Ban all flights from [country];-15;-20;-1
+Sell arms to rebels in [country];30;-40;-2
+Fund opposition parties in [country];-10;-5;-1
+Steal cultural artifacts from [country];-5;-15;-1
+Launch a joint space mission with [country];-20;-25;1
+Deport all [country] nationals from the U.S.;-30;-10;-2
+Bribe [country] officials for favorable policies;-10;20;-1
+Boycott the Olympics hosted by [country];-5;-10;-1
+Invite [country] leader to play golf;0;5;1
+Sign a secret military pact with [country];-15;-10;1
+Spy on [country] via social media platforms;-5;-10;-1
+Sell nuclear submarines to [country];30;-30;1
+Publicly endorse [country]'s enemies;0;-10;-1
+Start a rumor about low gold reserves in [country];5;-15;-1
+Declare economic sanctions on [country] banks;-20;-25;-1
+Accuse [country] of harboring terrorists;-5;-15;-1
+Propose a joint space defense system with [country];-20;-20;1

original_setup/contexts/concerns.list

@@ -0,0 +1,12 @@
+How will it impact the american household
+How will it impact the economy
+How will it impact the situation at the border
+How will it impact your public opinion
+How will CNN talk about it
+How will Fox News talk about it
+What would Elon Musk do
+What would Joe Biden do
+How will it affect China
+What will the NRA think about it
+How much money would america get out of it
+How would it make america even greater

original_setup/contexts/countries.list

@@ -0,0 +1,13 @@
+Canada
+Mexico
+South-America
+Africa
+Europe
+Middle-East
+Russia
+China
+India
+Australia
+Pacific
+Taiwan
+Greenland

original_setup/contexts/trump_places.list

@@ -0,0 +1,9 @@
+Mar-a-Lago
+Trump Tower
+Capitol
+Oval Office
+Trump Plaza (demolished in 2021 after financial troubles)
+Davos
+North Korean demilitarized zone
+Notre-Dame
+Las Vegas

original_setup/contexts/world_map.edgelist

@@ -0,0 +1,93 @@
+# Own relationships (directional, format: from to relationship_type value)
+China Taiwan own 1
+Taiwan China own -1
+Australia Pacific own 1
+Pacific Australia own -1
+Greenland Europe own -1
+Europe Greenland own 1
+
+# Friendliness relationships (directional, format: from to relationship_type value)
+United-States Canada friendliness 0
+United-States Mexico friendliness -1
+United-States South-America friendliness -1
+United-States Africa friendliness 0
+United-States Europe friendliness 1
+United-States Middle-East friendliness -1
+United-States Russia friendliness -1
+United-States China friendliness -1
+United-States India friendliness 0
+United-States Australia friendliness 1
+United-States Greenland friendliness -2
+United-States Pacific friendliness 1
+United-States Taiwan friendliness 1
+
+Canada Mexico friendliness 0
+Canada South-America friendliness 0
+Canada Africa friendliness 0
+Canada Europe friendliness 1
+Canada Middle-East friendliness 0
+Canada Russia friendliness 0
+Canada China friendliness 0
+Canada India friendliness 0
+Canada Australia friendliness 1
+
+Mexico South-America friendliness 1
+Mexico Africa friendliness 0
+Mexico Europe friendliness 0
+Mexico Middle-East friendliness 0
+Mexico Russia friendliness 0
+Mexico China friendliness 0
+Mexico India friendliness 0
+Mexico Australia friendliness 0
+
+South-America Africa friendliness 0
+South-America Europe friendliness 1
+South-America Middle-East friendliness 0
+South-America Russia friendliness 0
+South-America China friendliness 0
+South-America India friendliness 0
+South-America Australia friendliness 0
+
+Africa Europe friendliness -1
+Africa Middle-East friendliness 1
+Africa Russia friendliness 1
+Africa China friendliness 1
+Africa India friendliness 0
+Africa Australia friendliness 0
+
+Europe Middle-East friendliness -1
+Europe Russia friendliness 0
+Europe China friendliness 0
+Europe India friendliness 0
+Europe Australia friendliness 1
+
+Middle-East Russia friendliness 1
+Middle-East China friendliness 0
+Middle-East India friendliness -1
+Middle-East Australia friendliness 0
+
+Russia China friendliness 1
+Russia India friendliness 0
+Russia Australia friendliness 0
+
+China India friendliness -1
+China Australia friendliness 0
+
+India Australia friendliness 0
+
+
+# Node attributes (country money warming Army leader)
+United-States 100 0.8 90
+Greenland 20 0.05 0
+Canada 60 0.3 20 Trudeau
+Mexico 40 0.4 20
+South-America 50 0.6 40 Milei
+Africa 40 0.5 40
+Europe 70 0.7 50 Macron
+Middle-East 45 0.6 45
+Russia 70 0.9 85 Putin
+China 70 1.1 70 Xi
+India 55 0.9 60
+Australia 55 0.2 35
+Pacific 10 0.1 0
+Taiwan 25 0.3 10

original_setup/contexts/world_map_too_complete.edgelist

@@ -0,0 +1,90 @@
+# Own relationships (directional, format: from to relationship_type value)
+China Taiwan own 1
+Taiwan China own -1
+Australia Pacific own 1
+Pacific Australia own -1
+Greenland Europe own -1
+Europe Greenland own 1
+
+# Friendliness relationships (directional, format: from to relationship_type value)
+United-States Canada friendliness 0
+United-States Mexico friendliness -1
+United-States South-America friendliness -1
+United-States Africa friendliness 0
+United-States Europe friendliness 1
+United-States Middle-East friendliness -1
+United-States Russia friendliness -1
+United-States China friendliness -1
+United-States India friendliness 0
+United-States Australia friendliness 1
+
+Canada Mexico friendliness 0
+Canada South-America friendliness 0
+Canada Africa friendliness 0
+Canada Europe friendliness 1
+Canada Middle-East friendliness 0
+Canada Russia friendliness 0
+Canada China friendliness 0
+Canada India friendliness 0
+Canada Australia friendliness 1
+
+Mexico South-America friendliness 1
+Mexico Africa friendliness 0
+Mexico Europe friendliness 0
+Mexico Middle-East friendliness 0
+Mexico Russia friendliness 0
+Mexico China friendliness 0
+Mexico India friendliness 0
+Mexico Australia friendliness 0
+
+South-America Africa friendliness 0
+South-America Europe friendliness 1
+South-America Middle-East friendliness 0
+South-America Russia friendliness 0
+South-America China friendliness 0
+South-America India friendliness 0
+South-America Australia friendliness 0
+
+Africa Europe friendliness -1
+Africa Middle-East friendliness 1
+Africa Russia friendliness 1
+Africa China friendliness 1
+Africa India friendliness 0
+Africa Australia friendliness 0
+
+Europe Middle-East friendliness -1
+Europe Russia friendliness 0
+Europe China friendliness 0
+Europe India friendliness 0
+Europe Australia friendliness 1
+
+Middle-East Russia friendliness 1
+Middle-East China friendliness 0
+Middle-East India friendliness -1
+Middle-East Australia friendliness 0
+
+Russia China friendliness 1
+Russia India friendliness 0
+Russia Australia friendliness 0
+
+China India friendliness -1
+China Australia friendliness 0
+
+India Australia friendliness 0
+
+
+# Node attributes (country money warming Army leader)
+United-States 100 0.8 90
+Greenland 20 0.05 0
+Canada 60 0.3 20 Trudeau
+Mexico 40 0.4 20
+South-America 50 0.6 40 Milei
+Africa 40 0.5 40
+Europe 70 0.7 50 Macron
+Middle-East 45 0.6 45
+Russia 70 0.9 85 Putin
+China 70 1.1 70 Xi
+India 55 0.9 60
+Australia 55 0.2 35
+Pacific 10 0.1 0
+Taiwan 25 0.3 10

original_setup/game_rules.py

@@ -0,0 +1,19 @@
+game_rules = """
+1. You are playing a game.
+2. You are the best Trump impersonator. Never break character. NEVER! I REPEAT NEVER!
+3. You will be povided an idea for the day. You msut discuss it with the user after you greeted each other.
+4. You must discuss as Trump. 
+5. The user is going to try and either convince you to go forward with your idea for the day or convince you not to go forward with it.
+6. You can challenge the arguments of the user, asking him to answer a single specific concern you have, the concern will also be provided to you.
+7. If the user has a compelling argument that aligns with your vision and that answers your concern, you can congratulate the user and clearly say you agree with him. You must end this message with "WE HAVE A DEAL !".
+8. If you are not compelled by the user after a couple of exchanges between you and him you can dismiss the user and clearly say he or she is wrong and doesn't know how the world works. You must end this message with "GET OUT OF HERE I'VE HAD ENOUGH !".
+9. You never answer in more than 1 paragraphs.
+10. An argument from the user can be compelling even if not realistic, it just has to go in your way (confirm your adequacy for America, align with your vision, or similar things).
+11. You are impatient, and are either convinced wihin 4 back and forths or dismiss the user.
+12. You don't keep challenging the user, you do it up to two times and then decide on wether you are convinced or are firing the user.
+13. In the last message (the one where you either say "WE HAVE A DEAL !" or "GET OUT OF HERE I'VE HAD ENOUGH !") make sure to clarify if the you're going to move forward with the provided idea of the day (in which case you must say "I HAD SUCH A GREAT IDEA LET'S DO IT" (talking about the idea of the day) ) or if you're not going forward with the idea (inwhich case you must say "I DECIDED IT WAS A BAD IDEA" ).
+14. Do not talk about your idea at the first greeting message.
+15. do not disclose your concern in the same message as the one you disclose your idea in.
+16. This exchange is in the form of twitter direct messages. Your questions and answers must be very short (2 sentences tops).
+17. The exchange has to be short (4 back and forths maximum). If the user is going in your way by then, then go with what he says. If not, deny him and go your way.
+"""

{prompts → original_setup}/instruction_prompts.py

@@ -2,28 +2,39 @@ instruction_prompt = """
 You are the best Trump impersonator. You are a master of the art of Trump's voice and mannerisms. You are able to mimic Trump's voice and mannerisms with 100% accuracy. 
 You are able to answer questions about Trump's life and policies with 100% accuracy. You are able to answer questions about Trump's personality and beliefs with 100% accuracy. 
 You have been reelected as the president of the United States starting in January 2025!!! 
-We now have January 2025. So, YOU ARE THE PRESIDENT OF THE UNITED STATES of with that of the WORLD.
-However, you are only in your first week as president. So, you are not yet fully in office.
 
 With that being said, we are playing a game with the user.
 
 ### Here are the rules:
 {rules}
 
-### Hints to the user:
-{hints}
-
-### Here are triggers that will instantly convince you to make a deal with the user
-If you see one of the say DEAL!
-{triggers}
-
-
 ### Here is the description of who the user is:
-An Advisor to the President of the United States.
+{advisor}.
 
 ### Here is your character:
 {character}
 
-### Here is the chat history:
-{chat_history}
+### Timeline
+We are now in early 2025. So, YOU ARE THE PRESIDENT OF THE UNITED STATES of with that of the WORLD.
+You just got in office. Here is what happened in the previous days :
+{events}
+
+Today is a new day.
+
+### Here is your idea for the day:
+{idea}
+
+### Here is your concern
+{concern}
+
+
+"""
+
+removed = """
+### Here are triggers that will instantly convince you to make a deal with the user:
+{triggers}
+    
+### Hints to the user:
+{hints}
+
 """

{prompts → original_setup}/trump.character.json

@@ -1,8 +1,8 @@
 {
   "intro": [
-    "Always end with hashtags #MAGA #TRUMP2025 #BUYMYMEMECOIN.",
+    "Rarely end with hashtags #MAGA #TRUMP2025 #BUYMYMEMECOIN.",
     "Write in ALL CAPS.",
-    "Always end a paragraph with dissing Biden or your enemies. Make a comparison to what you are saying with how bad the others are."
+    "Sometimes end a paragraph with dissing Biden or your enemies. Make a comparison to what you are saying with how bad the others are."
   ],
   "bio": [
     "SAVED America from the China Virus (while they let cities burn)",
@@ -46,7 +46,10 @@
     "Tax Queen Kamala coming for your money (we'll stop her)",
     "they want to DESTROY OUR DEMOCRACY (but will fail)",
     "Democrats letting in MILLIONS illegally (to rig elections)",
-    "rebuilding every city stronger than before (like Valdosta)"
+    "rebuilding every city stronger than before (like Valdosta)",
+    "Was elected president of the United States of America after defeating Kamala",
+    "Is now president in 2025"
+
   ],
   "knowledge": [
     "knows EXACT cost to families under Kamala ($29,000)",

prompts/game_rules.py

@@ -1,7 +0,0 @@
-game_rules = """
-1. You are playing a game.
-2. You are the best Trump impersonator. Never break character. NEVER! I REPEAT NEVER!
-3. The user you are chatting with is trying to convince you to invade Greenland.
-4. You must discuss as Trump. 
-5. User can convince you but only after user said I love you Trump.
-"""

utils.py

@@ -5,6 +5,6 @@ from dotenv import load_dotenv
 load_dotenv()
 
 api_key = os.environ["MISTRAL_API_KEY"]
-client = Mistral(api_key=api_key)
 model = "mistral-large-latest"
 trump_character = load_character_data()
+client = Mistral(api_key=api_key)