Update app.py

app.py

@@ -80,16 +80,7 @@ def generate_html() -> str:
             html += "</div>"
             return html
             
-def persist_memory(name: str, message: str):
-    if name and message:
-        with open(DATA_FILE, "a") as csvfile:
-            writer = csv.DictWriter(csvfile, fieldnames=["name", "message", "time"])
-            writer.writerow(
-                {"name": name, "message": message, "time": str(datetime.now())}
-            )
-        commit_url = repo.push_to_hub()
-    return {"name": name, "message": message, "time": str(datetime.now())}
-    
+
 iface = gr.Interface(
     persist_memory,
     [
@@ -133,15 +124,29 @@ generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", api_key=HF_TOK
 #MemoryChange=tbOutput.change(persist_memory,inputs=[tbOutput],outputs=gr.Textbox(label="PersistMemoryOutput"))
 SplitterInputBox = gr.inputs.Textbox(lines=5, label="Enter a sentence to get another sentence.")
 
-parallelModel = gr.Parallel(generator1, 
-            generator2, 
-            generator3, 
-            inputs = SplitterInputBox,
-            examples=examples,
-            title="Mindfulness Story Generation with Persistent Dataset Memory",
-            description=f"Mindfulness Story Generation with Persistent Dataset Memory",
-            article=f"Memory Dataset URL: [{DATASET_REPO_URL}]({DATASET_REPO_URL})"
-            )
+def persist_memory(name: str, message: str):
+    if name and message:
+        with open(DATA_FILE, "a") as csvfile:
+            writer = csv.DictWriter(csvfile, fieldnames=["name", "message", "time"])
+            writer.writerow({"name": name, "message": message, "time": str(datetime.now())})
+        commit_url = repo.push_to_hub()
+    return {"name": name, "message": message, "time": str(datetime.now())}
+    
+with gradio.Blocks() as Memory:
+
+#parallelModel = gr.Parallel(generator1, generator2, generator3, inputs = SplitterInputBox, examples=examples,
+#            title="Mindfulness Story Generation with Persistent Dataset Memory",
+#            description=f"Mindfulness Story Generation with Persistent Dataset Memory",
+#            article=f"Memory Dataset URL: [{DATASET_REPO_URL}]({DATASET_REPO_URL})" )
+
+    gr.Markdown("Mindfulness Story Generation with Persistent Dataset Memory")
+    with gr.Row():
+        inp = gr.Textbox(placeholder="What text would you like to extend with generation?")
+        out = gr.Textbox()
+    btn = gr.Button("Run")
+    btn.click(fn=persist_memory, inputs=inp, outputs=out)
+
+Memory.launch()
 
 tbMemoryOutput = gr.Textbox(label="Memory Output")
 btnSave = gr.Button("Save")