!7

// class AIConcept comes from #1007160

concept ImportedFrom {

  S snippetID;

  long lastTime;

}

static JTable table;

static JCheckBox cbGrabImages;

volatile sbool grabImages;

volatile static double imageGrabberDelay = 1; // seconds, min 1

static Q uploadQ;

p {

  grabImages = headless();

  dbIndexing(AIConcept, 'searchedForSuggestedImage);

  framesBot(); // bot to activate my frames

  methodsBot("AI Concepts Import.", "importFrom", "importFromAI1Lol", "export");

  uploadQ = new Q(true);

  startImageGrabber();

  if (isHeadless()) ret;

  substance();

  swingMain();

  aiEnhancements();

}

svoid swingMain swing {

  showConceptsTable_postProcess.set(f reversedList);

  table = makeConceptsTable(AIConcept, func(AIConcept c) {

    litorderedmap(

      "Concept ID" := c.globalID,

      "Name/Description" := c.name,

      "Comments" := c.comment,

      "Has Image?" := c.pngFile.has(),

      //"Imported From" := c.importedFrom,

      "Originating Universe" := c.originatingUniverse,

      "Exportable" := c.exportable)

  });

  showFrame(makeTitle(), tableWithSearcher(table));

  onEnter(tableSearchField(table), f newConcept);

  onDoubleClick(table, r { editConcept(selectedConcept()) });

  addToWindow(table, withMargin(vstack(

    jRightAlignedLine(

      jbutton("New Concept...", f newConcept),

      tableDependButton(table, "Edit", r { editConcept(selectedConcept()) }),

      tableDependButton(table, "Duplicate", r { newConcept(selectedConcept()) }),

      jbutton("Export", f doExport),

      jbutton("Import...", f doImport)),

    withTopMargin(jRightAlignedLine(cbGrabImages = jCheckBox("Google For Images", grabImages))))));

  tablePopupMenu(table, voidfunc(JPopupMenu menu, final int row) {

    final S globalID = (S) getTableCell(table, row, 0);

    final AIConcept c = findConcept(AIConcept, +globalID);

    addMenuItem(menu, "Edit...", r { editConcept(c) });

    addMenuItem(menu, "Delete", r {

      withDBLock(r { deleteConcept(c) });

    });

    if (c.searchedForSuggestedImage)

      addMenuItem(menu, "Search for image again", r {

        deleteConcept(c.suggestedImage);

        cset(c, suggestedImage := null, searchedForSuggestedImage := false)

      });

  });

  tableEnableTextDrag(table);

  awtOnConceptChanges(table, 1000, r {

    setFrameTitle(table, makeTitle())

  });

  addAISuiteMenu(table, "AI");

  hideConsole();

}

svoid editConcept(final AIConcept c) {

  if (c == null) ret;

  final JTextField a = jTextField(c.name);

  final JTextField b = jTextField(c.comment);

  final JTextField cc = jTextField(c.globalID);

  final ImageChooser ic = new ImageChooser(c.pngFile.has() ? c.pngFile->getImageSafe() : null);

  ic.setTitleForUpload(c.name);

  PNGFile png = c.suggestedImage!;

  final ImageChooser ic2 = new ImageChooser(png != null ? png.getImageSafe() : null);

  final Var<S> md5 = new Var(md5OfPNGFile(png));

  awtOnConceptChanges(ic2, r {

    ic2.setTitleForUpload(c.name);

    PNGFile png = c.suggestedImage!;

    S newMD5 = md5OfPNGFile(png);

    if (png != null && neq(md5.get(), newMD5)) {

      md5.set(newMD5);

      ic2.setImage(png.getImageSafe());

    }

  }, true);

  Runnable r = r {

    BufferedImage image = ic.getImage();

    if (image == null && c.pngFile.has()) {

      //c.pngFile->delete();

      c.pngFile.clear();

      cset(c, imageMD5 := null);

    } else if (image != null &&

      (!c.pngFile.has() || !imagesIdentical(c.pngFile->getImage(), image))) {

      //if (c.pngFile.has()) c.pngFile->delete();

      setConceptImage(c, image);

    }

    cset(c, touched := now(),

      name := getTextTrim(a), comment := getTextTrim(b),

      globalID := getTextTrim(cc));

  };

  O[] params = {

    "Description/Name", a,

    "Comment (optional)", b,

    "Global ID", cc,

    "Example Image (optional)",

      westAndEast(ic, withLeftMargin(20, northAndCenter(vstack(

        jbutton("Accept", r {

        ic.setImage(ic2.getImage());

        findButton(getFrame(ic), "Save").doClick();

      }),

        jCenteredLabel("Suggested image:")), ic2))),

    r };

  //for i over params: print("#" + i + ": " + params[i]);

  renameSubmitButton("Save", showFormTitled("Edit concept " + c.globalID, params));

}

sS searcherInput() {

  ret getTextTrim(tableSearchField(table));

}

svoid newConcept { newConcept(null); }

svoid newConcept(final AIConcept c) {

  final JTextField a = jTextField(c == null ? searcherInput() : c.name);

  final JTextField b = jTextField(getString(c, "comment"));

  final JTextField cc = jTextField(aGlobalID());

  final JLabel lbl = jCenteredLabel("Suggested image:");

  final new ImageChooser ic;

  final new ImageChooser ic2;

  final new Var<S> md5;

  awtCalcRegularly(ic2, 500, 0, new Runnable {

    S lastTyped, lastSearched;

    public void run {

      S s = getTextTrim(a);

      if (nempty(s) && eq(s, lastTyped) && neq(s, lastSearched)) {

        lastSearched = s;

        lbl.setIcon(imageIcon(#1007255));

        try {

          final BufferedImage img = quickVisualize(s);

          if (img != null) swing { ic2.setImage(img); }

        } finally {

          lbl.setIcon(null);

        }

      }

      lastTyped = s;

    }

  });

  Runnable r = r {

    AIConcept c = cnew(AIConcept, touched := now(),

      name := getTextTrim(a), comment := getTextTrim(b), globalID := getTextTrim(cc));

    setConceptImage(c, ic.getImage());

  };

  renameSubmitButton("Create", showFormTitled("New concept",

    "Description/Name", a,

    "Comment (optional)", b,

    "Global ID", cc,

    "Example Image (optional)",

      westAndEast(ic, withLeftMargin(20, northAndCenter(vstack(

        jbutton("Accept", r { ic.setImage(ic2.getImage()) }),

        lbl), ic2))),

    r));

}

static AIConcept selectedConcept() {

  ret findConcept(AIConcept, globalID := selectedTableCell(table, 0));

}

svoid doImport {

  ImportedFrom last = last(sortedByField("lastTime", list(ImportedFrom)));

  final JTextField tf = jTextField(last == null ? #1007208 : last.snippetID);

  showFormTitled("Import A. I. Concepts",

    "Snippet ID", tf, r-thread {

      loading {

        importFrom(getTextTrim(tf));

      }

    });

}

svoid importFromAI1Lol {

  importFrom(freshConceptsFromAI1Lol(), "ai1.lol");

}

svoid importFrom(S snippetID)  {

  snippetID = fsI(snippetID);

  cset(uniq(ImportedFrom, +snippetID), lastTime := now());

  S text = loadSnippet(snippetID);

  Concepts newConcepts = new Concepts(snippetID).safeLoad(text);

  importFrom(newConcepts, snippetID);

}

svoid importFrom(Concepts newConcepts, S snippetID) {

  print(n(list(newConcepts, "AIConcept"), "concept") + " loaded from " + snippetID);

  Concepts myConcepts = mainConcepts;

  print("I have " + n(list(myConcepts, "AIConcept"), "concept");

  Map<S, AIConcept> myIDs = indexByField(list(myConcepts, AIConcept), "globalID");

  int nNew = 0;

  for (Concept c : list(newConcepts, "AIConcept")) {

    S id = getString(c, "globalID");

    bool has = myIDs.containsKey(id);

    print("ID: " + id + " - " + (has ? "I have it" : "I don't have it"));

    if (!has) {

      // Creating

      AIConcept c2 = cnew(AIConcept, importedFrom := snippetID, imported := now());

      importFields(c, c2);

      myIDs.put(id, c2);

      ++nNew;

    } else {

      // Updating

      AIConcept c2 = myIDs.get(id);

      if (c2.touched > c2.imported)

        print("  Concept touched locally! Not changing.");

      else {

        long time = now();

        cset(c2, imported := time, touched := time);

        int changes = importFields(c, c2);

        if (changes == 0)

          print("  No change.");

        else

          print("  Updated concept! " + n(changes, "fields") + " changed");

      }

    }

  }

  infoBox("Import done. " + n(nNew, "new concepts"));

}

static int importFields(Concept c, AIConcept c2) {

  c2.created = c.created;

  ret copyConceptFields(c, c2, "globalID", "name", "comment", "imageMD5", "originatingUniverse");

}

svoid doExport {

  swingExportConcepts("AI Concepts");

}

svoid export {

  swingExportConcepts_impl("AI Concepts");

}

sS makeTitle() {

  ret "My AI Concepts [" + countConcepts(AIConcept) + "]";

}

svoid startImageGrabber {

  thread "Image Grabber" {

    repeat with sleep imageGrabberDelay {

      if (headless() ? grabImages : isChecked(cbGrabImages)) {

        AIConcept c = findConcept(AIConcept, searchedForSuggestedImage := false);

        if (c != null) {

          cset(c, searchedForSuggestedImage := true);

          if (c.suggestedImage != null || c.importedFrom != null) continue;

          S query = c.name;

          //Window box = infoBox("Visualizing: " + c.name);

          BufferedImage img = quickVisualize(query);

          //disposeWindow(box);

          if (img != null) {

            showAnimation(img, c.name, 4);

            cset(c, suggestedImage := new PNGFile(quickVisualize_imagePath(query)), suggestedImageMD5 := md5OfBufferedImage(img));

          } else

            infoBox("No image found");

        }

      }

    }

  }

}

svoid setConceptImage(final AIConcept c, final BufferedImage img) {

  if (img == null) {

    c.pngFile.clear();

    cset(c, imageMD5 := null);

  } else {

    c.pngFile.set(new PNGFile(img));

    cset(c, imageMD5 := c.pngFile->md5());

    uploadQ.add(r { uploadToImageServerIfNotThere(img, aiConceptToString(c)) });

  }

}