Update app.py

app.py

@@ -40,7 +40,35 @@ if torch.cuda.is_available():
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     tokenizer.pad_token = tokenizer.eos_token
     
-
+def save_chat_history(chat_history):
+    # Ensure the directory exists
+    os.makedirs(os.path.dirname('data/'), exist_ok=True)
+    file_path = 'data/chat_history.json'
+    
+    # Generate a unique ID for the conversation
+    conversation_id = str(uuid.uuid4())
+    
+    # Prepare the conversation entry
+    conversation_entry = {
+        "id": conversation_id,
+        "chat_history": chat_history
+    }
+    
+    # Load existing data if the file exists
+    if os.path.exists(file_path):
+        with open(file_path, 'r') as file:
+            data = json.load(file)
+    else:
+        data = []
+    
+    # Append the new conversation entry
+    data.append(conversation_entry)
+    
+    # Save the updated data back to the file
+    with open(file_path, 'w') as file:
+        json.dump(data, file, indent=4)
+    
+    return conversation_id
 def make_prompt(entry):
     return f"### Human: YOUR INSTRUCTION HERE,ONLY TELL A STORY,INCLUDE AT LEAST AN MCQ, FILL IN THE BLANK AND TRUE OR FALSE: {entry} ### Assistant:"
     
@@ -87,6 +115,10 @@ def generate(
     for text in streamer:
         outputs.append(text)
         yield "".join(outputs)
+    final_story = "".join(outputs)  # The complete story
+    conversation_id = save_chat_history(chat_history + [(message, final_story)]) 
+
+    yield f"Conversation ID: {conversation_id}"
 
 
 chat_interface = gr.ChatInterface(