Update app.py

app.py

@@ -8,15 +8,13 @@ from fastapi.responses import HTMLResponse
 from datetime import datetime
 import random
 import string
-from datasets import Dataset
 import json
 
 # Set environment variables for GPU usage and memory allocation
 os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'expandable_segments:True'
 torch.cuda.empty_cache()
 torch.cuda.set_per_process_memory_fraction(0.8)  # Adjust the fraction as needed
-# Assuming you've created the file conversations.jsonl
-dataset = Dataset('json', data_files="articko/conversations/conversations.jsonl")
+
 # Initialize FastAPI application
 app = FastAPI(root_path="/api/v1")
 conversations = {}
@@ -283,10 +281,12 @@ async def chat_thread(request: Request, user_id: str):
         conversation_entry = {'thread_id': thread_id, 'message_id': message_id, 'user': msg_prompt, 'assistant': response}
 
         # Save conversation history to disk
-        save_conversation(user_id, conversation_entry)
+        conversation_thread = ({"user_id": user_id, "conversation": conversation})
+
+        
 
         # Return response and thread ID
-        return {'response': conversation_entry}
+        return {'response': conversation_thread}
     except HTTPException as e:
         raise e
     except Exception as e: