Make API and server compatible with OpenAI API (#1043)

* Make API and server compatible with OpenAI API
Modified encode_message and encode_dialog_prompt methods in generate.py to resolve streamlit errors.

* Make API and server compatible with OpenAI API
Added changes to generate.py and browser.py

Author: nobelchowdary

browser/browser.py

@@ -1,3 +1,4 @@
+import time
 import streamlit as st
 from openai import OpenAI
 

generate.py

@@ -56,7 +56,7 @@ def encode_header(self, role) -> List[int]:
         return tokens
 
     def encode_message(self, message) -> List[int]:
-        tokens = self.encode_header(message.role)
+        tokens = self.encode_header(message["role"]) 
         tokens.extend(
             self.tokenizer.encode(message["content"].strip(), bos=False, eos=False)
         )
@@ -69,7 +69,7 @@ def encode_dialog_prompt(self, dialog) -> List[int]:
         for message in dialog:
             tokens.extend(self.encode_message(message))
         # Add the start of an assistant message for the model to complete.
-        tokens.extend(self.encode_header({"role": "assistant", "content": ""}))
+        tokens.extend(self.encode_header("assistant"))  # Pass role directly as a string
         return tokens
 
 
@@ -905,3 +905,4 @@ def main(args):
     check_args(args, verb)
     args = arg_init(args)
     main(args)
+