From 2c9eb6f67a7c14d0588ccf114b474b8f23a732f0 Mon Sep 17 00:00:00 2001
From: robinrolle <robinrolle1748@gmail.com>
Date: Sat, 12 Apr 2025 16:37:06 +0200
Subject: [PATCH] fix imports

---
 app.py | 45 ++++++++++++++++++++++++++++++++++++---------
 1 file changed, 36 insertions(+), 9 deletions(-)

diff --git a/app.py b/app.py
index 2c6191e..f29ad56 100644
--- a/app.py
+++ b/app.py
@@ -1,24 +1,51 @@
-import logging
+from dto.requests import GameStartRequestDTO
+from services.extractor import run_extraction_chain
+from services.julius_baer_api_client import JuliusBaerApiClient
+from validation.from_passport import FromPassport
 
-from flask import Flask
-
-import config
-from dto.requests import GameStartRequestDTO, GameDecisionRequestDTO
 from services.player import Player
+from utils.parsers import process_passport
+from flask import Flask
+import config
 
 app = Flask(__name__)
 
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - [%(module)s] - %(message)s')
-
 
 @app.route('/')
-def hello_world():  # put application's code here
+def hello_world():
     return 'Hello World!'
 
-
 if __name__ == '__main__':
+    jb_client = JuliusBaerApiClient()
+    game_start_request = GameStartRequestDTO(player_name=config.API_TEAM)
+    res = jb_client.start_game(game_start_request)
+
+
+    passport_data = res.client_data.get("passport")
+
+    prompt_template = (
+        "Extract the following information from the provided passport text.\n"
+        "Return only JSON matching this format:\n{format_instructions}\n\n"
+        "Pay special attention to the passport number and signature.\n"
+        "Passport text:\n{processed_text}"
+    )
+
+    result = run_extraction_chain(
+        raw_file_data=passport_data,
+        file_processor=process_passport,
+        pydantic_model=FromPassport,
+        prompt_template=prompt_template,
+    )
+
+    print(result)
 
     player = Player()
     player.play_on_separate_thread()
 
     app.run()
+
+    # res.session_id
+    # UUID('fde19363-a3d5-432e-8b87-54a6dd54f0dd')
+    # second test UUID('e3d58302-400a-4bc6-9772-ae50de43c9f4')
+    # UUID('f8b2a0a6-d4e0-45e6-900f-8ecb3c28f993')
+    # UUID('f8b2a0a6-d4e0-45e6-900f-8ecb3c28f993')
\ No newline at end of file