csvai 0.1.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,751 @@
1
+ Metadata-Version: 2.4
2
+ Name: csvai
3
+ Version: 0.1.0
4
+ Summary: Enrich CSV or Excel rows using OpenAI models.
5
+ Author: Zyxware Technologies, Vimal Joseph
6
+ License: GNU GENERAL PUBLIC LICENSE
7
+ Version 2, June 1991
8
+
9
+ Copyright (C) 1989, 1991 Free Software Foundation, Inc.,
10
+ 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
11
+ Everyone is permitted to copy and distribute verbatim copies
12
+ of this license document, but changing it is not allowed.
13
+
14
+ Preamble
15
+
16
+ The licenses for most software are designed to take away your
17
+ freedom to share and change it. By contrast, the GNU General Public
18
+ License is intended to guarantee your freedom to share and change free
19
+ software--to make sure the software is free for all its users. This
20
+ General Public License applies to most of the Free Software
21
+ Foundation's software and to any other program whose authors commit to
22
+ using it. (Some other Free Software Foundation software is covered by
23
+ the GNU Lesser General Public License instead.) You can apply it to
24
+ your programs, too.
25
+
26
+ When we speak of free software, we are referring to freedom, not
27
+ price. Our General Public Licenses are designed to make sure that you
28
+ have the freedom to distribute copies of free software (and charge for
29
+ this service if you wish), that you receive source code or can get it
30
+ if you want it, that you can change the software or use pieces of it
31
+ in new free programs; and that you know you can do these things.
32
+
33
+ To protect your rights, we need to make restrictions that forbid
34
+ anyone to deny you these rights or to ask you to surrender the rights.
35
+ These restrictions translate to certain responsibilities for you if you
36
+ distribute copies of the software, or if you modify it.
37
+
38
+ For example, if you distribute copies of such a program, whether
39
+ gratis or for a fee, you must give the recipients all the rights that
40
+ you have. You must make sure that they, too, receive or can get the
41
+ source code. And you must show them these terms so they know their
42
+ rights.
43
+
44
+ We protect your rights with two steps: (1) copyright the software, and
45
+ (2) offer you this license which gives you legal permission to copy,
46
+ distribute and/or modify the software.
47
+
48
+ Also, for each author's protection and ours, we want to make certain
49
+ that everyone understands that there is no warranty for this free
50
+ software. If the software is modified by someone else and passed on, we
51
+ want its recipients to know that what they have is not the original, so
52
+ that any problems introduced by others will not reflect on the original
53
+ authors' reputations.
54
+
55
+ Finally, any free program is threatened constantly by software
56
+ patents. We wish to avoid the danger that redistributors of a free
57
+ program will individually obtain patent licenses, in effect making the
58
+ program proprietary. To prevent this, we have made it clear that any
59
+ patent must be licensed for everyone's free use or not licensed at all.
60
+
61
+ The precise terms and conditions for copying, distribution and
62
+ modification follow.
63
+
64
+ GNU GENERAL PUBLIC LICENSE
65
+ TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION
66
+
67
+ 0. This License applies to any program or other work which contains
68
+ a notice placed by the copyright holder saying it may be distributed
69
+ under the terms of this General Public License. The "Program", below,
70
+ refers to any such program or work, and a "work based on the Program"
71
+ means either the Program or any derivative work under copyright law:
72
+ that is to say, a work containing the Program or a portion of it,
73
+ either verbatim or with modifications and/or translated into another
74
+ language. (Hereinafter, translation is included without limitation in
75
+ the term "modification".) Each licensee is addressed as "you".
76
+
77
+ Activities other than copying, distribution and modification are not
78
+ covered by this License; they are outside its scope. The act of
79
+ running the Program is not restricted, and the output from the Program
80
+ is covered only if its contents constitute a work based on the
81
+ Program (independent of having been made by running the Program).
82
+ Whether that is true depends on what the Program does.
83
+
84
+ 1. You may copy and distribute verbatim copies of the Program's
85
+ source code as you receive it, in any medium, provided that you
86
+ conspicuously and appropriately publish on each copy an appropriate
87
+ copyright notice and disclaimer of warranty; keep intact all the
88
+ notices that refer to this License and to the absence of any warranty;
89
+ and give any other recipients of the Program a copy of this License
90
+ along with the Program.
91
+
92
+ You may charge a fee for the physical act of transferring a copy, and
93
+ you may at your option offer warranty protection in exchange for a fee.
94
+
95
+ 2. You may modify your copy or copies of the Program or any portion
96
+ of it, thus forming a work based on the Program, and copy and
97
+ distribute such modifications or work under the terms of Section 1
98
+ above, provided that you also meet all of these conditions:
99
+
100
+ a) You must cause the modified files to carry prominent notices
101
+ stating that you changed the files and the date of any change.
102
+
103
+ b) You must cause any work that you distribute or publish, that in
104
+ whole or in part contains or is derived from the Program or any
105
+ part thereof, to be licensed as a whole at no charge to all third
106
+ parties under the terms of this License.
107
+
108
+ c) If the modified program normally reads commands interactively
109
+ when run, you must cause it, when started running for such
110
+ interactive use in the most ordinary way, to print or display an
111
+ announcement including an appropriate copyright notice and a
112
+ notice that there is no warranty (or else, saying that you provide
113
+ a warranty) and that users may redistribute the program under
114
+ these conditions, and telling the user how to view a copy of this
115
+ License. (Exception: if the Program itself is interactive but
116
+ does not normally print such an announcement, your work based on
117
+ the Program is not required to print an announcement.)
118
+
119
+ These requirements apply to the modified work as a whole. If
120
+ identifiable sections of that work are not derived from the Program,
121
+ and can be reasonably considered independent and separate works in
122
+ themselves, then this License, and its terms, do not apply to those
123
+ sections when you distribute them as separate works. But when you
124
+ distribute the same sections as part of a whole which is a work based
125
+ on the Program, the distribution of the whole must be on the terms of
126
+ this License, whose permissions for other licensees extend to the
127
+ entire whole, and thus to each and every part regardless of who wrote it.
128
+
129
+ Thus, it is not the intent of this section to claim rights or contest
130
+ your rights to work written entirely by you; rather, the intent is to
131
+ exercise the right to control the distribution of derivative or
132
+ collective works based on the Program.
133
+
134
+ In addition, mere aggregation of another work not based on the Program
135
+ with the Program (or with a work based on the Program) on a volume of
136
+ a storage or distribution medium does not bring the other work under
137
+ the scope of this License.
138
+
139
+ 3. You may copy and distribute the Program (or a work based on it,
140
+ under Section 2) in object code or executable form under the terms of
141
+ Sections 1 and 2 above provided that you also do one of the following:
142
+
143
+ a) Accompany it with the complete corresponding machine-readable
144
+ source code, which must be distributed under the terms of Sections
145
+ 1 and 2 above on a medium customarily used for software interchange; or,
146
+
147
+ b) Accompany it with a written offer, valid for at least three
148
+ years, to give any third party, for a charge no more than your
149
+ cost of physically performing source distribution, a complete
150
+ machine-readable copy of the corresponding source code, to be
151
+ distributed under the terms of Sections 1 and 2 above on a medium
152
+ customarily used for software interchange; or,
153
+
154
+ c) Accompany it with the information you received as to the offer
155
+ to distribute corresponding source code. (This alternative is
156
+ allowed only for noncommercial distribution and only if you
157
+ received the program in object code or executable form with such
158
+ an offer, in accord with Subsection b above.)
159
+
160
+ The source code for a work means the preferred form of the work for
161
+ making modifications to it. For an executable work, complete source
162
+ code means all the source code for all modules it contains, plus any
163
+ associated interface definition files, plus the scripts used to
164
+ control compilation and installation of the executable. However, as a
165
+ special exception, the source code distributed need not include
166
+ anything that is normally distributed (in either source or binary
167
+ form) with the major components (compiler, kernel, and so on) of the
168
+ operating system on which the executable runs, unless that component
169
+ itself accompanies the executable.
170
+
171
+ If distribution of executable or object code is made by offering
172
+ access to copy from a designated place, then offering equivalent
173
+ access to copy the source code from the same place counts as
174
+ distribution of the source code, even though third parties are not
175
+ compelled to copy the source along with the object code.
176
+
177
+ 4. You may not copy, modify, sublicense, or distribute the Program
178
+ except as expressly provided under this License. Any attempt
179
+ otherwise to copy, modify, sublicense or distribute the Program is
180
+ void, and will automatically terminate your rights under this License.
181
+ However, parties who have received copies, or rights, from you under
182
+ this License will not have their licenses terminated so long as such
183
+ parties remain in full compliance.
184
+
185
+ 5. You are not required to accept this License, since you have not
186
+ signed it. However, nothing else grants you permission to modify or
187
+ distribute the Program or its derivative works. These actions are
188
+ prohibited by law if you do not accept this License. Therefore, by
189
+ modifying or distributing the Program (or any work based on the
190
+ Program), you indicate your acceptance of this License to do so, and
191
+ all its terms and conditions for copying, distributing or modifying
192
+ the Program or works based on it.
193
+
194
+ 6. Each time you redistribute the Program (or any work based on the
195
+ Program), the recipient automatically receives a license from the
196
+ original licensor to copy, distribute or modify the Program subject to
197
+ these terms and conditions. You may not impose any further
198
+ restrictions on the recipients' exercise of the rights granted herein.
199
+ You are not responsible for enforcing compliance by third parties to
200
+ this License.
201
+
202
+ 7. If, as a consequence of a court judgment or allegation of patent
203
+ infringement or for any other reason (not limited to patent issues),
204
+ conditions are imposed on you (whether by court order, agreement or
205
+ otherwise) that contradict the conditions of this License, they do not
206
+ excuse you from the conditions of this License. If you cannot
207
+ distribute so as to satisfy simultaneously your obligations under this
208
+ License and any other pertinent obligations, then as a consequence you
209
+ may not distribute the Program at all. For example, if a patent
210
+ license would not permit royalty-free redistribution of the Program by
211
+ all those who receive copies directly or indirectly through you, then
212
+ the only way you could satisfy both it and this License would be to
213
+ refrain entirely from distribution of the Program.
214
+
215
+ If any portion of this section is held invalid or unenforceable under
216
+ any particular circumstance, the balance of the section is intended to
217
+ apply and the section as a whole is intended to apply in other
218
+ circumstances.
219
+
220
+ It is not the purpose of this section to induce you to infringe any
221
+ patents or other property right claims or to contest validity of any
222
+ such claims; this section has the sole purpose of protecting the
223
+ integrity of the free software distribution system, which is
224
+ implemented by public license practices. Many people have made
225
+ generous contributions to the wide range of software distributed
226
+ through that system in reliance on consistent application of that
227
+ system; it is up to the author/donor to decide if he or she is willing
228
+ to distribute software through any other system and a licensee cannot
229
+ impose that choice.
230
+
231
+ This section is intended to make thoroughly clear what is believed to
232
+ be a consequence of the rest of this License.
233
+
234
+ 8. If the distribution and/or use of the Program is restricted in
235
+ certain countries either by patents or by copyrighted interfaces, the
236
+ original copyright holder who places the Program under this License
237
+ may add an explicit geographical distribution limitation excluding
238
+ those countries, so that distribution is permitted only in or among
239
+ countries not thus excluded. In such case, this License incorporates
240
+ the limitation as if written in the body of this License.
241
+
242
+ 9. The Free Software Foundation may publish revised and/or new versions
243
+ of the General Public License from time to time. Such new versions will
244
+ be similar in spirit to the present version, but may differ in detail to
245
+ address new problems or concerns.
246
+
247
+ Each version is given a distinguishing version number. If the Program
248
+ specifies a version number of this License which applies to it and "any
249
+ later version", you have the option of following the terms and conditions
250
+ either of that version or of any later version published by the Free
251
+ Software Foundation. If the Program does not specify a version number of
252
+ this License, you may choose any version ever published by the Free Software
253
+ Foundation.
254
+
255
+ 10. If you wish to incorporate parts of the Program into other free
256
+ programs whose distribution conditions are different, write to the author
257
+ to ask for permission. For software which is copyrighted by the Free
258
+ Software Foundation, write to the Free Software Foundation; we sometimes
259
+ make exceptions for this. Our decision will be guided by the two goals
260
+ of preserving the free status of all derivatives of our free software and
261
+ of promoting the sharing and reuse of software generally.
262
+
263
+ NO WARRANTY
264
+
265
+ 11. BECAUSE THE PROGRAM IS LICENSED FREE OF CHARGE, THERE IS NO WARRANTY
266
+ FOR THE PROGRAM, TO THE EXTENT PERMITTED BY APPLICABLE LAW. EXCEPT WHEN
267
+ OTHERWISE STATED IN WRITING THE COPYRIGHT HOLDERS AND/OR OTHER PARTIES
268
+ PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED
269
+ OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
270
+ MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE ENTIRE RISK AS
271
+ TO THE QUALITY AND PERFORMANCE OF THE PROGRAM IS WITH YOU. SHOULD THE
272
+ PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF ALL NECESSARY SERVICING,
273
+ REPAIR OR CORRECTION.
274
+
275
+ 12. IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING
276
+ WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MAY MODIFY AND/OR
277
+ REDISTRIBUTE THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES,
278
+ INCLUDING ANY GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING
279
+ OUT OF THE USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED
280
+ TO LOSS OF DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY
281
+ YOU OR THIRD PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER
282
+ PROGRAMS), EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE
283
+ POSSIBILITY OF SUCH DAMAGES.
284
+
285
+ END OF TERMS AND CONDITIONS
286
+
287
+ How to Apply These Terms to Your New Programs
288
+
289
+ If you develop a new program, and you want it to be of the greatest
290
+ possible use to the public, the best way to achieve this is to make it
291
+ free software which everyone can redistribute and change under these terms.
292
+
293
+ To do so, attach the following notices to the program. It is safest
294
+ to attach them to the start of each source file to most effectively
295
+ convey the exclusion of warranty; and each file should have at least
296
+ the "copyright" line and a pointer to where the full notice is found.
297
+
298
+ <one line to give the program's name and a brief idea of what it does.>
299
+ Copyright (C) <year> <name of author>
300
+
301
+ This program is free software; you can redistribute it and/or modify
302
+ it under the terms of the GNU General Public License as published by
303
+ the Free Software Foundation; either version 2 of the License, or
304
+ (at your option) any later version.
305
+
306
+ This program is distributed in the hope that it will be useful,
307
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
308
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
309
+ GNU General Public License for more details.
310
+
311
+ You should have received a copy of the GNU General Public License along
312
+ with this program; if not, write to the Free Software Foundation, Inc.,
313
+ 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
314
+
315
+ Also add information on how to contact you by electronic and paper mail.
316
+
317
+ If the program is interactive, make it output a short notice like this
318
+ when it starts in an interactive mode:
319
+
320
+ Gnomovision version 69, Copyright (C) year name of author
321
+ Gnomovision comes with ABSOLUTELY NO WARRANTY; for details type `show w'.
322
+ This is free software, and you are welcome to redistribute it
323
+ under certain conditions; type `show c' for details.
324
+
325
+ The hypothetical commands `show w' and `show c' should show the appropriate
326
+ parts of the General Public License. Of course, the commands you use may
327
+ be called something other than `show w' and `show c'; they could even be
328
+ mouse-clicks or menu items--whatever suits your program.
329
+
330
+ You should also get your employer (if you work as a programmer) or your
331
+ school, if any, to sign a "copyright disclaimer" for the program, if
332
+ necessary. Here is a sample; alter the names:
333
+
334
+ Yoyodyne, Inc., hereby disclaims all copyright interest in the program
335
+ `Gnomovision' (which makes passes at compilers) written by James Hacker.
336
+
337
+ <signature of Ty Coon>, 1 April 1989
338
+ Ty Coon, President of Vice
339
+
340
+ This General Public License does not permit incorporating your program into
341
+ proprietary programs. If your program is a subroutine library, you may
342
+ consider it more useful to permit linking proprietary applications with the
343
+ library. If this is what you want to do, use the GNU Lesser General
344
+ Public License instead of this License.
345
+
346
+ Project-URL: Homepage, https://www.zyxware.com/article/6935/csvai-automate-data-enrichment-any-csv-or-excel-file-generative-ai
347
+ Project-URL: Bug Tracker, https://github.com/zyxware/csvai/issues
348
+ Project-URL: Source, https://github.com/zyxware/csvai
349
+ Project-URL: Contact, https://www.zyxware.com/contact-us
350
+ Keywords: csv,excel,ai,openai,data enrichment,llm,automation
351
+ Classifier: Programming Language :: Python :: 3
352
+ Classifier: License :: OSI Approved :: GNU General Public License v2 (GPLv2)
353
+ Classifier: Operating System :: OS Independent
354
+ Requires-Python: >=3.9
355
+ Description-Content-Type: text/markdown
356
+ License-File: LICENSE
357
+ Requires-Dist: openai>=1.99.0
358
+ Requires-Dist: jinja2>=3.1.4
359
+ Requires-Dist: python-dotenv>=1.0.1
360
+ Requires-Dist: pandas>=2.0.0
361
+ Requires-Dist: openpyxl>=3.1.0
362
+ Provides-Extra: ui
363
+ Requires-Dist: streamlit; extra == "ui"
364
+ Dynamic: license-file
365
+
366
+ # CSVAI — Apply an AI prompt to each row in a CSV or Excel file and write enriched results
367
+
368
+ The `csvai` library reads an input CSV or Excel file, renders a prompt for each row (you can use raw column names like `{{ Address }}`), calls an **OpenAI model via the Responses API**, and writes the original columns plus AI-generated fields to an output CSV or Excel file.
369
+
370
+ The tool is **async + concurrent**, **resumable**, and **crash-safe**. It supports **Structured Outputs** with a **JSON Schema** for reliable JSON, or **JSON mode** (without a schema) if you prefer a lighter setup.
371
+
372
+ We also have a **CSV AI Prompt Builder** (a Custom GPT) to help you generate prompts and JSON Schemas tailored to your CSVs.
373
+
374
+ ---
375
+
376
+ ## Features
377
+
378
+ * **Structured Outputs**: enforce exact JSON with a schema for consistent, validated results.
379
+ * **JSON mode**: force a single JSON object without defining a schema.
380
+ * **Async & concurrent**: process many rows in parallel for faster throughput.
381
+ * **Resumable**: rows already written (by `id`) are skipped on re-run.
382
+ * **CSV or Excel**: handle `.csv` and `.xlsx` inputs and outputs.
383
+
384
+ ---
385
+
386
+ ## Installation
387
+
388
+ Requires Python **3.9+**.
389
+ OpenAI API Key: Create a key - https://platform.openai.com/api-keys and use it in the .env file with OPENAI_API_KEY=
390
+ See example.env in the [project repo](https://github.com/zyxware/csvai).
391
+
392
+ ### From PyPI
393
+
394
+ ```bash
395
+ pip install csvai
396
+ # Include Streamlit UI dependencies
397
+ pip install "csvai[ui]"
398
+ ```
399
+
400
+ ### From GitHub
401
+
402
+ ```bash
403
+ # Install directly from the repository
404
+ pip install git+https://github.com/zyxware/csvai.git
405
+ # With Streamlit UI dependencies
406
+ pip install "csvai[ui] @ git+https://github.com/zyxware/csvai.git"
407
+ ```
408
+
409
+ ### For local development
410
+
411
+ ```bash
412
+ git clone https://github.com/zyxware/csvai
413
+ cd csvai
414
+ python -m venv .venv
415
+ source .venv/bin/activate
416
+ pip install -r requirements.txt
417
+ pip install -e .
418
+ cp example.env .env
419
+ # Edit .env and set OPENAI_API_KEY=sk-...
420
+ ```
421
+
422
+ Installing the package exposes the `csvai` CLI and the `csvai-ui` command.
423
+
424
+ ---
425
+
426
+ ## Usage
427
+
428
+ ### CLI
429
+
430
+ #### Auto-discovery
431
+
432
+ If you name your files like this:
433
+
434
+ ```
435
+ input.csv # or input.xlsx
436
+ input.prompt.txt
437
+ input.schema.json # optional
438
+ ```
439
+
440
+ Run:
441
+
442
+ ```bash
443
+ csvai input.csv # or input.xlsx
444
+ ```
445
+
446
+ #### Or specify prompt & schema explicitly
447
+
448
+ ```bash
449
+ # With a prompt and a strict schema (best reliability)
450
+ csvai address.xlsx --prompt address.prompt.txt --schema address.schema.json
451
+
452
+ # Or JSON mode (no schema; still a single JSON object)
453
+ csvai address.xlsx --prompt address.prompt.txt
454
+ ```
455
+
456
+ Sample datasets (`address.csv` and `address.xlsx`) with the matching prompt and schema live in the `example/` directory.
457
+
458
+ ### Streamlit UI
459
+
460
+ After installing with the `ui` extra, launch the web interface:
461
+
462
+ ```bash
463
+ csvai-ui
464
+ ```
465
+
466
+ The UI lets you upload a CSV/Excel file, provide a prompt and optional schema.
467
+
468
+ ---
469
+
470
+ ## Example Prompt & Schema
471
+
472
+ ### Prompt (`address.prompt.txt`)
473
+
474
+ ```text
475
+ Extract city, state, and country from the given address.
476
+
477
+ Rules:
478
+ - city: city/town/locality (preserve accents, proper case)
479
+ - state: ISO-standard name of the state/region/province or "" if none
480
+ - country: ISO 3166 English short name of the country; infer if obvious, else ""
481
+ - Ignore descriptors like "(EU)"
482
+ - Do not guess street-level info
483
+
484
+ Inputs:
485
+ Address: {{Address}}
486
+ ```
487
+
488
+ ### Schema (`address.schema.json`)
489
+
490
+ ```json
491
+ {
492
+ "type": "object",
493
+ "properties": {
494
+ "city": {
495
+ "type": "string",
496
+ "description": "City, town, or locality name with correct casing and accents preserved"
497
+ },
498
+ "state": {
499
+ "type": "string",
500
+ "description": "ISO-standard name of the state, region, or province, or empty string if none"
501
+ },
502
+ "country": {
503
+ "type": "string",
504
+ "description": "ISO 3166 English short name of the country, inferred if obvious, else empty string"
505
+ }
506
+ },
507
+ "required": ["city", "state", "country"],
508
+ "additionalProperties": false
509
+ }
510
+ ```
511
+ ---
512
+
513
+ ## Creating Prompts & Schemas with the CSV AI Prompt Builder
514
+
515
+ You can use the **[CSV AI Prompt Builder](https://chat.openai.com/g/g-689d8067bd888191a896d2cfdab27a39-csv-ai-prompt-builder)** custom GPT to:
516
+
517
+ * Quickly design a **prompt** tailored to your CSV data.
518
+ * Generate a **JSON Schema** that matches your desired structured output.
519
+
520
+ **Example input to the builder:**
521
+
522
+ ```
523
+ File: reviews.csv. Inputs: title,body,stars. Output: sentiment,summary.
524
+ ```
525
+
526
+ **Example result:**
527
+
528
+ **Prompt**
529
+
530
+ ```
531
+ Analyze each review and produce sentiment and a concise summary.
532
+
533
+ Rules:
534
+ - sentiment: one of positive, neutral, negative.
535
+ - Star mapping: stars ≤ 2 ⇒ negative; 3 ⇒ neutral; ≥ 4 ⇒ positive. If stars is blank or invalid, infer from tone.
536
+ - summary: 1–2 sentences, factual, include key pros/cons, no emojis, no first person, no marketing fluff.
537
+ - Use the same language as the Body.
538
+ - Return only the fields required by the tool schema.
539
+
540
+ Inputs:
541
+ Title: {{title}}
542
+ Body: {{body}}
543
+ Stars: {{stars}}
544
+ ```
545
+
546
+ **Schema**
547
+
548
+ ```json
549
+ {
550
+ "type": "object",
551
+ "properties": {
552
+ "sentiment": {
553
+ "type": "string",
554
+ "description": "Overall sentiment derived from stars and/or tone; one of positive, neutral, negative",
555
+ "enum": ["positive", "neutral", "negative"]
556
+ },
557
+ "summary": {
558
+ "type": "string",
559
+ "description": "Concise 1–2 sentence summary capturing key pros/cons without opinionated fluff"
560
+ }
561
+ },
562
+ "required": ["sentiment", "summary"],
563
+ "additionalProperties": false
564
+ }
565
+ ```
566
+
567
+ **Command to execute**
568
+
569
+ ```bash
570
+ python -m csvai.cli reviews.csv --prompt reviews.prompt.txt --schema reviews.schema.json
571
+ ```
572
+ **Tip — have the builder generate a schema for you**
573
+ * **I have `products.csv` with Product Title, Product Description, Category, and Sub Category. Help me enrich with SEO meta fields.**
574
+ * **I have `reviews.csv` with Title, Body, and Stars. Help me extract sentiment and generate a short summary.**
575
+ * **I have `address.csv` with an Address field. Help me extract City, State, and Country using ISO-standard names.**
576
+ * **I have `tickets.csv` with Subject and Description. Help me classify each ticket into predefined support categories.**
577
+ * **I have `posts.csv` with Title, Body, URL, Image URL, Brand, and Platform. Help me generate social media captions, hashtags, emojis, CTAs, and alt text.**
578
+ * **I have `jobs.csv` with Job Title and Description. Help me categorize jobs into sectors and identify the level of seniority.**
579
+
580
+
581
+ ---
582
+
583
+ ## CLI
584
+
585
+ ```bash
586
+ csvai INPUT.csv [--prompt PROMPT_FILE] [--output OUTPUT_FILE]
587
+ [--limit N] [--model MODEL] [--schema SCHEMA_FILE]
588
+ ```
589
+
590
+ **Flags**
591
+
592
+ * `--prompt, -p` — path to a plaintext prompt file (Jinja template).
593
+ * `--output, -o` — output CSV path (default: `<input>_enriched.csv`).
594
+ * `--limit` — process only the first `N` new/pending rows.
595
+ * `--model` — model name (default from `.env`, falls back to `gpt-4o-mini`).
596
+ * `--schema` — path to a JSON Schema for structured outputs (optional).
597
+
598
+ ---
599
+
600
+ ## Environment Variables (`.env`)
601
+
602
+ See [`example.env`](example.env) for all configurable variables.
603
+
604
+ ```ini
605
+ OPENAI_API_KEY=sk-...
606
+ DEFAULT_MODEL=gpt-4o-mini
607
+ MAX_OUTPUT_TOKENS=600
608
+ TEMPERATURE=0.2
609
+ MAX_CONCURRENT_REQUESTS=12
610
+ PROCESSING_BATCH_SIZE=100
611
+ REQUEST_TIMEOUT=45
612
+ ALT_PROMPT_SUFFIX=.prompt.txt
613
+ OUTPUT_FILE_SUFFIX=_enriched.csv
614
+ ```
615
+
616
+ ---
617
+
618
+ ## Input/Output Behavior
619
+
620
+ * **Input CSV**: the script reads all rows. If an `id` column exists, it’s used to resume. If not, rows are indexed `0..N-1` internally for this run.
621
+ * **Prompt rendering**: every row is sanitized so `{{ Raw Header }}` becomes `{{ Raw_Header }}`. You can also reference the raw values as `{{ raw["Raw Header"] }}` if needed.
622
+ * **Output CSV**: contains the original columns plus AI-generated fields. The **header is fixed** after the first successful batch; later rows are written with the same header order.
623
+ * **Resume**: rerunning skips rows whose `id` is already present in the output file.
624
+
625
+ ---
626
+
627
+ ## Structured Outputs vs JSON Mode
628
+
629
+ ### Structured Outputs (recommended)
630
+
631
+ When you pass `--schema`, the request includes:
632
+
633
+ ```python
634
+ text={
635
+ "format": {
636
+ "type": "json_schema",
637
+ "name": "row_schema",
638
+ "schema": schema,
639
+ "strict": true
640
+ }
641
+ }
642
+ ```
643
+
644
+ This guarantees the model returns **exactly** the keys/types you expect.
645
+
646
+ ### JSON Mode (no schema)
647
+
648
+ When no schema is provided, the request includes:
649
+
650
+ ```python
651
+ text={"format": {"type": "json_object"}}
652
+ ```
653
+
654
+ The model must still return a single JSON object, but no exact schema is enforced.
655
+
656
+ > **Prompting tip:** mention the word **JSON** in your prompt and explicitly list the expected fields to improve compliance in JSON mode.
657
+
658
+ ---
659
+
660
+ ## Performance & Concurrency
661
+
662
+ * Concurrency is controlled by `MAX_CONCURRENT_REQUESTS`.
663
+ * Increase gradually; too high can trigger API rate limits.
664
+ * `PROCESSING_BATCH_SIZE` controls how many results are written per batch.
665
+ * `REQUEST_TIMEOUT` guards slow requests; the script retries with backoff.
666
+
667
+ ---
668
+
669
+ ## Troubleshooting
670
+
671
+ **“Missing required parameter: `text.format.name`”**
672
+ You used structured outputs but didn’t include `name` **alongside** `type` and `schema`. The script already sends this correctly; ensure you’re on the latest version and that `--schema` points to the right file.
673
+
674
+ **“Invalid schema … `required` must include every key”**
675
+ The Responses structured-outputs path expects `required` to include **all** keys in `properties`. Either (a) add them all to `required`, (b) remove non-required keys from `properties`, or (c) use JSON mode.
676
+
677
+ **Rows not resuming**
678
+ Ensure there’s an `id` column in both input and output. If not present, the script uses positional IDs for the current run only.
679
+
680
+ ---
681
+
682
+ ## FAQ
683
+
684
+ **Q: Does it run concurrently?**
685
+ Yes. Concurrency is controlled via `MAX_CONCURRENT_REQUESTS` (default 10).
686
+
687
+ **Q: Can I rely on an `id` column?**
688
+ Yes. If present in the input CSV, it’s used for resumability. Otherwise rows are indexed for the session.
689
+
690
+ **Q: Can I output nested JSON?**
691
+ The schema can be nested, but CSV is flat. If you want nested data, extend the script with a flattener (e.g., convert `address.street` → `address_street`).
692
+
693
+ **Q: Which models work?**
694
+ Recent `gpt-4o*` models support Responses + Structured Outputs. If a model doesn’t support it, use JSON mode.
695
+
696
+ **Q: Do I need a JSON schema?**
697
+ No, but it’s strongly recommended for stable columns and fewer parse failures.
698
+
699
+ ---
700
+
701
+ ## Support
702
+
703
+ This application was developed as an internal tool and we will continue to improve and optimize it as long as we use it. If you would like us to customize this or build a similar or related system to automate your tasks with AI, we are available for **commercial support**.
704
+
705
+ ---
706
+
707
+ ### About Zyxware Technologies
708
+
709
+ At **Zyxware Technologies**, our mission is to help organizations harness the power of technology to solve real-world problems. Guided by our founding values of honesty and fairness, we are committed to delivering genuine value to our clients and the free and open-source community.
710
+
711
+ **CSVAI** is a direct result of this philosophy. We originally developed it to automate and streamline our own internal data-enrichment tasks. Realizing its potential to help others, we are sharing it as a free tool in the spirit of our commitment to Free Software.
712
+
713
+ Our expertise is centered around our **AI & Automation Services**. We specialize in building intelligent solutions that reduce manual effort, streamline business operations, and unlock data-driven insights. While we provide powerful free tools like this one, we also offer **custom development and commercial support** for businesses that require tailored AI solutions.
714
+
715
+ If you're looking to automate a unique business process or build a similar system, we invite you to [**reach out to us**](https://www.zyxware.com/contact-us) to schedule a free discovery call.
716
+
717
+ ---
718
+
719
+ ## Updates
720
+
721
+ For updates and new versions, visit: [Project Page @ Zyxware](https://www.zyxware.com/article/6935/csvai-automate-data-enrichment-any-csv-or-excel-file-generative-ai)
722
+
723
+ ---
724
+
725
+ ## Contact
726
+
727
+ [https://www.zyxware.com/contact-us](https://www.zyxware.com/contact-us)
728
+
729
+ ---
730
+
731
+ ## Source Repository
732
+
733
+ [https://github.com/zyxware/csvai](https://github.com/zyxware/csvai)
734
+
735
+ ---
736
+
737
+ ## Reporting Issues
738
+
739
+ [https://github.com/zyxware/csvai/issues](https://github.com/zyxware/csvai/issues)
740
+
741
+ ---
742
+
743
+ ## License and Disclaimer
744
+
745
+ **GPL v2** – Free to use & modify. Use it at your own risk. We are not collecting any user data.
746
+
747
+ ---
748
+
749
+ ## Need Help or Commercial Support?
750
+
751
+ If you have any questions, feel free to [contact us](https://www.zyxware.com/contact-us).