From 4a97cc70416beafa2a333a5b78a3765371a88b61 Mon Sep 17 00:00:00 2001
From: Gene Kogan <kogan.gene@gmail.com>
Date: Thu, 28 Mar 2024 14:09:04 -0700
Subject: [PATCH] naming

---
 app/generator.py | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/app/generator.py b/app/generator.py
index faaba27..6130f0b 100644
--- a/app/generator.py
+++ b/app/generator.py
@@ -1,8 +1,12 @@
+import re
 import os
 import uuid
 import requests
 from fastapi import BackgroundTasks
 
+from .scenarios.tasks import general_assistant
+from .models.tasks import SimpleAssistantRequest
+
 from .models import (
     MonologueRequest,
     DialogueRequest,
@@ -51,6 +55,24 @@
 ]
 
 
+def name_interaction(prompt: str):
+    naming_prompt = f"Take the following text prompt and output a short phrase or sentence with 5-15 words which captures the essence of the text. It will be used to give a name to the interaction. If the text given is already a short phrase, just output that. Output *only* the short phrase. Do not include pretext, restatement, or other extraneous text.\n\n---\n\nText:\n {prompt}"
+    print("----")
+    print(naming_prompt)
+    request = SimpleAssistantRequest(
+        prompt=naming_prompt,
+        model="gpt-3.5-turbo",
+        params={"temperature": 0.0, "max_tokens": 100},
+    )
+    try:
+        name = general_assistant(request)
+    except Exception as e:
+        name = re.sub(r"\W+", " ", prompt).replace("\n", "").replace("\r", "")[:100]
+    print(name)
+    print("----")
+    return name
+
+
 def process_task(task_id: str, request: TaskRequest):
     print("config", request.config)
 
@@ -255,7 +277,7 @@ def send_progress_update(progress: float):
         output = TaskResult(
             files=[output_url],
             thumbnails=[thumbnail_url],
-            name=prompt,
+            name=name_interaction(prompt),
             attributes={},
             progress=1,
             isFinal=True,