Update app.py

app.py

@@ -42,29 +42,6 @@ if torch.cuda.is_available():
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     tokenizer.pad_token = tokenizer.eos_token
     
-def save_chat_history(chat_history):
-    
-    file_path = 'chat_history.json'
-    
-    conversation_id = str(uuid.uuid4())
-    
-    conversation_entry = {
-        "id": conversation_id,
-        "chat_history": chat_history
-    }
-    
-    if os.path.exists(file_path):
-        with open(file_path, 'r') as file:
-            data = json.load(file)
-    else:
-        data = []
-    
-    data.append(conversation_entry)
-    
-    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: {entry} ### Assistant:"
     
@@ -111,11 +88,6 @@ 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(
     fn=generate,