Upload app.py

app.py

@@ -0,0 +1,110 @@
+import os
+import tempfile
+import gradio as gr
+import toml
+from aphra import translate
+
+theme = gr.themes.Soft(
+    primary_hue="rose",
+    secondary_hue="pink",
+    spacing_size="lg",
+)
+
+def create_config_file(api_key, writer_model, searcher_model, critic_model):
+    config = {
+        "openrouter": {"api_key": api_key},
+        "llms": {
+            "writer": writer_model,
+            "searcher": searcher_model,
+            "critiquer": critic_model
+        }
+    }
+    with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.toml') as tmp:
+        toml.dump(config, tmp)
+    return tmp.name
+
+def process_input(file, text_input, api_key, writer_model, searcher_model, critic_model, source_lang, target_lang):
+    if file is not None:
+        with open(file, 'r', encoding='utf-8') as file:
+            text = file.read()
+    else:
+        text = text_input
+    config_file = create_config_file(api_key, writer_model, searcher_model, critic_model)
+    try:
+        translation = translate(
+            source_language=source_lang,
+            target_language=target_lang,
+            text=text,
+            config_file=config_file,
+            log_calls=False
+        )
+    finally:
+        os.unlink(config_file)
+    
+    return translation
+
+def create_interface():
+    with gr.Blocks(theme=theme) as demo:
+        gr.Markdown("<font size=6.5><center>🌐💬 Aphra</center></font>")
+        gr.Markdown(
+            """<div style="display: flex;align-items: center;justify-content: center">
+            [<a href="https://davidlms.github.io/aphra/">Project Page</a>] | [<a href="https://github.com/DavidLMS/aphra">Github</a>]</div>
+            """
+        )
+        gr.Markdown("🌐💬 Aphra is an open-source translation agent designed to enhance the quality of text translations by leveraging large language models (LLMs).")
+        
+        with gr.Row():
+            api_key = gr.Textbox(label="Openrouter API Key", type="password")
+            
+            writer_model = gr.Dropdown(
+                ["anthropic/claude-3.5-sonnet:beta", "openai/gpt-4o-2024-08-06", "google/gemini-pro-1.5-exp"],
+                label="Writer Model",
+                value="anthropic/claude-3.5-sonnet:beta",
+                allow_custom_value=True
+            )
+            searcher_model = gr.Dropdown(
+                ["perplexity/llama-3-sonar-large-32k-online", "perplexity/llama-3.1-sonar-huge-128k-online", "perplexity/llama-3.1-sonar-small-128k-online"],
+                label="Searcher Model",
+                value="perplexity/llama-3-sonar-large-32k-online",
+                allow_custom_value=True
+            )
+            critic_model = gr.Dropdown(
+                ["anthropic/claude-3.5-sonnet:beta", "openai/gpt-4o-2024-08-06", "google/gemini-pro-1.5-exp"],
+                label="Critic Model",
+                value="anthropic/claude-3.5-sonnet:beta",
+                allow_custom_value=True
+            )
+
+        with gr.Row():
+            source_lang = gr.Dropdown(
+                ["Spanish", "English", "French", "German"],
+                label="Source Language",
+                value="Spanish",
+                allow_custom_value=True
+            )
+            target_lang = gr.Dropdown(
+                ["English", "Spanish", "French", "German"],
+                label="Target Language",
+                value="English",
+                allow_custom_value=True
+            )
+
+        with gr.Row(): 
+            file = gr.File(label="Upload .txt or .md file", file_types=[".txt", ".md"])
+            text_input = gr.Textbox(label="Or paste your text here", lines=5)
+        
+        translate_btn = gr.Button("Translate with 🌐💬 Aphra")
+        
+        output = gr.Textbox(label="Translation by 🌐💬 Aphra")
+        
+        translate_btn.click(
+            process_input,
+            inputs=[file, text_input, api_key, writer_model, searcher_model, critic_model, source_lang, target_lang],
+            outputs=[output]
+        )
+    
+    return demo
+
+if __name__ == "__main__":
+    interface = create_interface()
+    interface.launch()