Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| from model import GeneralModel | |
| import os | |
| import chess | |
| import chess.engine | |
| import stat | |
| import re | |
| def app(): | |
| # Creating an object of prediction service | |
| pred = GeneralModel() | |
| api_key = st.sidebar.text_input("APIkey get it here https://beta.openai.com/account/api-keys", type="password") | |
| # Using the streamlit cache | |
| def eval(fenstring): | |
| output = "" | |
| os.chmod("./stockfish_14_x64_popcnt",0o0777) | |
| engine = chess.engine.SimpleEngine.popen_uci("./stockfish_14_x64_popcnt") | |
| # Score: PovScore(Cp(+20), WHITE) | |
| board = chess.Board(fenstring) | |
| info = engine.analyse(board, chess.engine.Limit(depth=20),multipv=3) | |
| # Score: PovScore(Mate(+1), WHITE) | |
| engine.quit() | |
| evalstring = str(info) | |
| print (evalstring) | |
| #hacky way to parse best move, need to imprpve | |
| substrings = evalstring.split('Move.from_uci') | |
| # Extract the text between the single quotes using a regular expression | |
| match = re.search(r"'(.*?)'", substrings[1]) | |
| topmove = match.group(1) | |
| print(topmove) | |
| # Create a list of lists, where each inner list contains the UCI strings for a single move | |
| stringprompt = 'Based on the chess board position at this fen string ' + fenstring + 'can you give an in depth analysis as to why ' + topmove + ' is the best move to play based on Material, Development, Center-control, King-safety and Pawn-structures if relevant. Also describe any tactics or ideas that we should be thinking about' | |
| return pred.model_prediction(input=stringprompt.strip() , api_key=api_key) | |
| def process_prompt(input): | |
| return pred.model_prediction(input=input.strip() , api_key=api_key) | |
| if api_key: | |
| # Setting up the Title | |
| st.title("Stockfish fen eval to GPT-3 top move explanation chess coach") | |
| # st.write("---") | |
| s_example = "r2qnrk1/1p1bbppp/p1np4/2p2p2/P1B1P3/2NP1NQ1/1PP3PP/R1B2RK1 w - - 0 12" | |
| input = st.text_area( | |
| "input a fen string and get a gpt 3.5 explanation of the top move", | |
| value=s_example, | |
| max_chars=1250, | |
| height=50, | |
| ) | |
| if st.button("Submit"): | |
| with st.spinner(text="In progress"): | |
| report_text = eval(input) | |
| st.markdown(report_text) | |
| else: | |
| st.error("🔑 Please enter API Key") |