adding all the changes

app.py

@@ -1,14 +1,7 @@
 from io import StringIO
 import streamlit as st
 from loguru import logger
-
-
-class CoreClient:
-
-    def generate(self, source: str, target: str, code: str):
-        logger.info(f"translate {source} into {target}, with SQL {code}")
-        return code
-
+from client import OpenAIService, SQLService, FacebookLLAMAService, GoogleT5Service
 
 st.set_page_config(
     page_title=' A SQL Generative Pre-trained Transformer',
@@ -19,7 +12,9 @@ st.set_page_config(
 databases = ['Oracle', 'SQLServer', 'MySQL', 'DB2', 'PostgreSQL', 'Snowflake', 'Redshift']
 # -------------
 
-st.sidebar.header('🎃 A SQL Transformer for Migration ')
+st.sidebar.header('🎃 A SQL Transformer for Migration')
+
+model = st.sidebar.radio('Model', ['openai-text-david-003', 'google-t5', 'facebook-llama'])
 
 source_database = st.sidebar.selectbox(
     label='📕 Source Database',
@@ -43,24 +38,31 @@ input_file = st.sidebar.file_uploader(
     label="	📄 Choose a SQL file",
     accept_multiple_files=False)
 
-client = CoreClient()
-
 
 def transform():
+    client: SQLService = None
+    if model == "openai-text-david-003":
+        client = OpenAIService(st.secrets("OPEN-ORG"), st.secrets("OPEN-KEY"))
+    elif model == "google-t5":
+        client = GoogleT5Service()
+    elif model == "facebook-llama":
+        client = FacebookLLAMAService()
+    logger.info(f"Using Model:{model}")
+
     code = input_text
     source = source_database
     target = target_database
     if code:
-        solution = client.generate(source, target, code)
-        st.code(solution, language='sql')
+        solutions = client.translate(source, target, code)
+        st.code(solutions[0], language='sql')
     else:
         if input_file is not None:
             # To convert to a string based IO:
             stringio = StringIO(input_file.getvalue().decode("utf-8"))
             # To read file as string:
             sql = stringio.read()
-            solution = client.generate(source, target, sql)
-            st.code(solution, language='sql')
+            solutions = client.translate(source, target, sql)
+            st.code(solutions[0], language='sql')
 
 
 # ---------------------------------------

client.py

@@ -0,0 +1,64 @@
+from abc import ABC, abstractmethod
+from typing import List
+from loguru import logger
+import openai
+
+
+class SQLService(ABC):
+
+    @abstractmethod
+    def translate(self, source_db: str, target_db: str, sql: str) -> List:
+        ...
+
+
+class OpenAIService(SQLService):
+
+    def __init__(self, organization, api_key) -> None:
+        super().__init__()
+        openai.organization = organization
+        openai.api_key = api_key
+
+    def translate(self, source_db: str, target_db: str, sql: str) -> List:
+        results = []
+        try:
+            response = openai.Completion.create(
+                model="text-davinci-003",
+                prompt=f"##### Translate this function  from Oracle into Postgresql\n"
+                       f"### {source_db}"
+                       f""
+                       f"   {sql}"
+                       f""
+                       f"### {target_db}",
+                temperature=0,
+                max_tokens=2048,
+                top_p=1,
+                frequency_penalty=0,
+                presence_penalty=0,
+                stop=["###"]
+            )
+            for choice in response.choices:
+                logger.info(f"transform {source_db} to {target_db}, SQL:")
+                logger.debug(choice.text)
+                results.append(choice.text)
+        except Exception as ex:
+            logger.error(f"transform from {source_db} to {target_db}, failed \n {sql}")
+            logger.exception(ex)
+        return results
+
+
+class GoogleT5Service(SQLService):
+
+    def treanslate(self, source_db: str, target_db: str, sql: str) -> List:
+        pass
+
+    def __init__(self):
+        ...
+
+
+class FacebookLLAMAService(SQLService):
+
+    def treanslate(self, source_db: str, target_db: str, sql: str) -> List:
+        pass
+
+    def __init__(self):
+        ...