Update main.py

main.py

@@ -25,16 +25,16 @@ pipe.safety_checker = dummy
 def hello():
     return "Hello, I'm Artist"
 
-async def generate(prompt):
-    result = await Linlada._create_completion(model='gpt-4', messages=[
-                                     {"role": "user", "content": promp}], stream=False) # alterative model setting
+@app.post('/generate_completion')
+async def generate_completion(
+    model: str = Query('gpt-4', description='The model to use for generating the completion'),
+    messages: List[Dict[str, str]] = Query(..., description='The list of messages to generate the completion for'),
+    stream: bool = Query(False, description='Whether to stream the response')
+):
+    response = index._create_completion(model=model, messages=messages, stream=stream)
+    
+    result = []
     for message in response:
-    return message
-
-@app.get("/gen/{prompt}")
-def generate_image(prompt: str):
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-    result = loop.run_until_complete(generate(prompt))
-    loop.close()
+        result.append(message)
+    
     return result