Update main.py

main.py

@@ -1,5 +1,6 @@
 import Linlada
 from fastapi import FastAPI
+import asyncio
 from fastapi.middleware.cors import CORSMiddleware
 
 app = FastAPI()
@@ -16,7 +17,19 @@ app.add_middleware(  # add the middleware
 def hello():
     return "Hello, I'm Artist"
 
-@app.post("/generate/{message}")
-def generate_completion(message: str):
-    response = Linlada._create_completion(model='gpt-4', messages=[{"role": 'user', "content": message}], stream=False)
-    return response
+async def generate(prompt):
+        result = index._create_completion(model='gpt-4', messages=[
+                                        {"role": "user", "content": prompt}], stream=True)
+        chat = []                            
+        for message in result:
+            chat.append(message)
+        sentence = ''.join(chat)        
+        return sentence    
+
+@app.get('/linlada/{prompt}')
+def generate_image_route(prompt: str):
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    result = loop.run_until_complete(generate(prompt))
+    loop.close()
+    return result