Spaces:
Sleeping
Sleeping
File size: 3,165 Bytes
46c25c8 42972b8 46c25c8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
import os
import json
from dotenv import load_dotenv
from groq import Groq
import re
def review_chess_game(pgn_file_path):
# Load environment variables and initialize Groq client
load_dotenv()
API_KEY = os.getenv("GROQ_API_KEY")
if not API_KEY:
raise ValueError("API key not found. Please set GROQ_API_KEY in the .env file.")
client = Groq(api_key=API_KEY)
# JSON structured prompt
template = (
"You are tasked with reviewing a chess game in PGN format: {pgn_content}. "
"Please provide the analysis in JSON format with the following structure:\n"
"{{\n"
' "summary": "Brief game summary",\n'
' "move_reviews": [\n'
' {{"move": "e4", "evaluation": "Good", "commentary": "Solid central control"}},\n'
' {{"move": "d5", "evaluation": "Brilliant", "commentary": "Aggressive center contest"}}\n'
' ],\n'
' "biggest_blunders": {{\n'
' "player1": "Qxb7",\n'
' "player2": "None"\n'
' }},\n'
' "recommendations": {{\n'
' "player1": "Focus on central control",\n'
' "player2": "Continue aggressive play"\n'
' }}\n'
"}}\n"
"Make sure the JSON is well-formatted and does not contain any invalid content."
)
# Read PGN content
try:
with open(pgn_file_path, 'r') as file:
pgn_content = file.read()
except FileNotFoundError:
return {"error": f"File '{pgn_file_path}' not found."}
# Format the prompt
prompt = template.format(pgn_content=pgn_content)
# Interact with Groq API
try:
completion = client.chat.completions.create(
model="llama-3.3-70b-versatile",
messages=[{"role": "user", "content": prompt}],
temperature=1,
max_tokens=4096,
top_p=1,
stream=True,
stop=None,
)
response_text = ""
for message in completion:
if message.choices and message.choices[0].delta.content:
response_text += message.choices[0].delta.content
response_text = response_text.strip()
# Remove unnecessary text before JSON starts
response_text = re.sub(r"(?s)^.*?\{", "{", response_text).strip()
response_text = re.sub(r"```json|```", "", response_text).strip()
# Ensure the response is valid JSON
try:
structured_data = json.loads(response_text)
return structured_data
except json.JSONDecodeError as e:
return {"error": f"Failed to parse JSON: {str(e)}", "raw_response": response_text}
except Exception as e:
return {"error": f"Error processing PGN: {str(e)}"}
def validate_json(review):
"""Check if the input is a valid JSON string or dictionary."""
if isinstance(review, dict):
print("Valid JSON (dictionary)")
return True # Already a valid Python dictionary
try:
json.loads(review) # Attempt to parse JSON string
print("Valid JSON (string)")
return True
except json.JSONDecodeError:
print("Invalid JSON response")
return False # Return False if JSON is invalid
|