Update app.py

app.py

@@ -48,9 +48,8 @@ if (!window.hasBeenRun) {
 UseMemory=True
 
 
-DATASET_REPO_URL="https://huggingface.co/datasets/awacke1/ChatbotMemory.csv"
-DATASET_REPO_ID="awacke1/ChatbotMemory.csv"
-DATA_FILENAME="ChatbotMemory.csv"
+DATASET_REPO_URL = "https://huggingface.co/datasets/awacke1/BookComposer"
+DATA_FILENAME = "BookComposer.csv"
 DATA_FILE=os.path.join("data", DATA_FILENAME)
 HF_TOKEN=os.environ.get("HF_TOKEN")
 
@@ -72,7 +71,6 @@ def get_df(name: str):
     dataset = load_dataset(str, split="train")
     return dataset
     
-#def store_message(name: str, message: str) -> str:
 def store_message(name: str, message: str):
     if name and message:
         with open(DATA_FILE, "a") as csvfile:
@@ -81,16 +79,7 @@ def store_message(name: str, message: str):
                 {"time": str(datetime.now()), "message": message.strip(), "name": name.strip()  }
             )
         commit_url = repo.push_to_hub()
-
-        # test api retrieval of any dataset that is saved, then return it...
-        # app = FastAPI()
-        # see: https://gradio.app/sharing_your_app/#api-page
-
-        # f=get_df(DATASET_REPO_ID)
-        # print(f)
-    #return commit_url
     return ""
-# ----------------------------------------------- For Memory
     
 
 repo = Repository(
@@ -98,27 +87,6 @@ repo = Repository(
 )
 
 
-def generate_html() -> str:
-    with open(DATA_FILE) as csvfile:
-        reader = csv.DictReader(csvfile)
-        rows = []
-        for row in reader:
-            rows.append(row)
-        rows.reverse()
-        if len(rows) == 0:
-            return "no messages yet"
-        else:
-            html = "<div class='chatbot'>"
-            for row in rows:
-                html += "<div>"
-                html += f"<span>{row['name']}</span>"
-                html += f"<span class='message'>{row['message']}</span>"
-                html += "</div>"
-            html += "</div>"
-            return html
-
-    return generate_html()
-