Create app.py

app.py

@@ -0,0 +1,64 @@
+import os
+import gradio as gr
+
+import google.auth
+from google.auth.transport.requests import Request
+
+def get_google_token():
+    credentials, project = google.auth.load_credentials_from_dict(
+        json.loads(os.environ.get('GCP_FINETUNE_KEY')),
+        scopes=[
+            "https://www.googleapis.com/auth/cloud-platform",
+            "https://www.googleapis.com/auth/generative-language.tuning",
+        ],
+    )
+    request = Request()
+    credentials.refresh(request)
+    access_token = credentials.token
+    return access_token
+
+
+def dubpro_english_transliteration(text):
+    API_URL = os.environ.get("GEMINI_FINETUNED_HINDI_ENG_API")
+    BEARER_TOKEN = get_google_token()
+    headers = {
+        "Authorization": f"Bearer {BEARER_TOKEN}",
+        "Content-Type": "application/json",
+    }
+    payload = {
+        "contents": [
+            {
+                "parts": [{"text": f"input: {text}"}],
+                "role": "user",
+            }
+        ],
+        "generationConfig": {
+            "maxOutputTokens": 8192,
+            "temperature": 0.85,
+        },
+        "safetySettings": [
+            {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
+            {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
+            {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
+            {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
+        ],
+    }
+    result = requests.post(
+        url=API_URL,
+        headers=headers,
+        json=payload
+    )
+    response = result.json()
+    response_content = response['candidates'][0]['content']['parts'][0]['text'].replace("output:", "").strip().replace("'text':", "").replace("{", "").replace("}", "").strip().strip("'").strip('"')
+    return response_content
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("English Transliteration Tool")
+    with gr.Row():
+        input_text = gr.Textbox(label="Input text", info="Please enter English text.")
+        output_text = gr.Textbox(label="Output text")
+    submit = gr.Button("Submit")
+    submit.click(dubpro_english_transliteration, input_text, output_text)
+
+demo.launch()