seed_db.py

import json
import requests

# The URL
URL = "http://127.0.0.1:8000/pairs"

def seed_database():
    try:
        with open("translation_pairs.jsonl", "r") as f:
            for line in f:
                # Parse each line of the JSONL file
                data = json.loads(line)
                
                # Format
                payload = {
                    "source_language": data["source_language"],
                    "target_language": data["target_language"],
                    "sentence": data["sentence"],
                    "translation": data["translation"]
                }
                
                # -> API
                response = requests.post(URL, json=payload)
                if response.status_code == 200:
                    print(f"Added: {data['sentence'][:30]}...")
                else:
                    print(f"Failed to add: {data['sentence'][:30]}")
    except FileNotFoundError:
        print("Error: translation_pairs.jsonl not found. Make sure it's in this folder.")

if __name__ == "__main__":
    seed_database()