handle api key for save cost limit api

LanguageTranslator/model/openai_model.py

@@ -20,7 +20,7 @@ class Model:
 class OpenAIModel(Model):
     def __init__(self, model_name: str, api_key: str):
         self.model_name = model_name
-        openai.api_key = api_key
+        # openai.api_key = api_key
 
     def make_translation_request(self, prompt):
         attempts = 0

LanguageTranslator/utils/constants.py

@@ -0,0 +1,13 @@
+import os
+import openai
+
+
+def set_api_key(api_key):
+    openai.api_key = api_key
+    return api_key
+
+
+def stop_api_key(value):
+    os.environ.pop('OPENAI_API_KEY', None)
+    openai.api_key = None
+    return 'Stopped API key'

api.py

@@ -3,6 +3,7 @@ from starlette.responses import RedirectResponse
 from typing import Union, List
 from pydantic import BaseModel
 from translator import ServerTranslator
+from LanguageTranslator.utils.constants import set_api_key
 import json
 import uvicorn
 
@@ -28,13 +29,16 @@ async def index():
 @app.post("/translate", response_model=TranslationResult)
 async def run_translation_manual(
         text: str = Query(..., description="Input text to translate"),
-        dest_language: str = Query(..., description="Destination language")):
+        dest_language: str = Query(..., description="Destination language"),
+        api_key: str = Query(..., description="OpenAI API Key")):
+    set_api_key(api_key)
     # Splitting the input text
     text = text.split(',')
     # Creating and processing the translator
     processing_language = ServerTranslator.language_translator(
         text=text,
         dest_language=dest_language,
+        openai_api_key=api_key
     )
     # Getting the translated result
     result_response = processing_language.translate()
@@ -42,14 +46,16 @@ async def run_translation_manual(
 
 
 @app.post("/translate_json", response_model=TranslationResult)
-async def run_translation_auto(json_file: UploadFile):
+async def run_translation_auto(json_file: UploadFile,  api_key: str = Query(..., description="OpenAI API Key")):
     try:
+        set_api_key(api_key)
         # Reading the JSON content from the file
         json_content = await json_file.read()
         json_data = json.loads(json_content.decode("utf-8"))
         # Creating and processing the translator
         processing_language = ServerTranslator.language_translator(
-            json_data
+            json_data,
+            openai_api_key=api_key
         )
         # Getting the translated result
         result_response = processing_language.translate()

app.py

@@ -1,8 +1,14 @@
 import gradio as gr
 from translator import ServerTranslator
+from LanguageTranslator.utils.constants import set_api_key, stop_api_key
 import json
 
 
+def apply_api_key(api_key):
+    api_key = set_api_key(api_key=api_key)
+    return f'Successfully set {api_key}'
+
+
 def clear_all(input_json, input_text, input_dest_lang, translated_text_json, translted_text_text):
     return None, "", "", "", ""
 
@@ -63,5 +69,22 @@ with gr.Blocks() as demo:
         [input_json, input_text, input_dest_lang, translated_text_json, translated_text_text],
         [input_json, input_text, input_dest_lang, translated_text_json, translated_text_text]
     )
+    with gr.Row():
+        with gr.Column():
+            status = gr.Textbox(label='API KEY STATUS')
+            api_key_set = gr.Textbox(label='Set your OPENAI API KEY')
+            api_key_set_button = gr.Button("Set API key")
+            api_key_set_button.click(
+                apply_api_key,
+                [api_key_set],
+                [status]
+            )
+        with gr.Column():
+            status_2 = gr.Textbox(label='STOP API KEY STATUS')
+            stop_api_button = gr.Button('Stop API key')
+            stop_api_button.click(
+                stop_api_key,
+                [],
+                [status_2])
 
 demo.launch(share=True)