Update app.py

app.py

@@ -4,13 +4,7 @@ from typing import Dict, Any
 from functools import partial
 
 from flask import Flask, request, jsonify
-from langchain import PromptTemplate, LLMChain
-from langchain.llms import OpenAI
-from langchain.chains import ConversationChain
-from langchain.memory import ConversationBufferMemory
-from langchain.vectorstores import Chroma
-from langchain.embeddings import OpenAIEmbeddings
-from langchain.document_loaders import TextLoader
+from transformers import pipeline
 
 logging.basicConfig(level=logging.INFO)
 
@@ -51,10 +45,7 @@ class UIUXWizard:
     def __init__(self, model_manager: ModelManager, vector_store=None):
         self.model_manager = model_manager
         self.vector_store = vector_store
-        self.conversation_chain = ConversationChain(
-            llm=OpenAI(temperature=0.7),
-            memory=ConversationBufferMemory(),
-        )
+        self.conversation_chain = pipeline("text-generation", model="gpt2")
 
     async def start(self):
         await self.model_manager.start()
@@ -70,7 +61,8 @@ class UIUXWizard:
             return "\n".join(results)
 
     def get_conversation_response(self, query):
-        return self.conversation_chain.run(query)
+        response = self.conversation_chain(query, max_length=5000, num_return_sequences=1)[0]['generated_text']
+        return response
 
 # Define VersionControl class
 class VersionControl:
@@ -216,7 +208,7 @@ if __name__ == "__main__":
 
     # Start the cluster and task processing loop
     asyncio.run(cluster.start())
-    asyncio.run(cluster.process_tasks())
+    asyncio.create_task(cluster.process_tasks())
 
     # Run Flask app
-    app.run(debug=True)
+    app.run(debug=True)