Update app.py

app.py

@@ -9,6 +9,7 @@ from PIL import Image
 from streamlit_image_comparison import image_comparison
 import numpy as np
 import re
+import cohere
 #import chromadb
 
 from textwrap import dedent
@@ -378,6 +379,153 @@ with tab2:
                     st.error("Please enter some text to convert.")
 
 
+        cohere_api_key = os.environ.get('COHERE_API_KEY')  # Fetch the API key from environment variable
+        
+        if cohere_api_key is None:
+            st.error("API key not found. Please set the COHERE_API_KEY environment variable.")
+            st.stop()
+        
+        # Get API Key Here - https://dashboard.cohere.com/api-keys
+        
+        co = cohere.Client(cohere_api_key)  # Use the fetched API key
+        
+        def generate_text(prompt, model='c4ai-aya', max_tokens=300, temperature=0.4):
+            response = co.generate(
+                model=model,
+                prompt=prompt,
+                max_tokens=max_tokens,
+                temperature=temperature,
+                k=0,
+                stop_sequences=[],
+                return_likelihoods='NONE')
+            return response.generations[0].text
+        
+        # Streamlit interface
+        st.title("Text Generation with Cohere")
+        
+        lang_id = {
+            "Afrikaans": "af",
+            "Amharic": "am",
+            "Arabic": "ar",
+            "Asturian": "ast",
+            "Azerbaijani": "az",
+            "Bashkir": "ba",
+            "Belarusian": "be",
+            "Bulgarian": "bg",
+            "Bengali": "bn",
+            "Breton": "br",
+            "Bosnian": "bs",
+            "Catalan": "ca",
+            "Cebuano": "ceb",
+            "Czech": "cs",
+            "Welsh": "cy",
+            "Danish": "da",
+            "German": "de",
+            "Greeek": "el",
+            "English": "en",
+            "Spanish": "es",
+            "Estonian": "et",
+            "Persian": "fa",
+            "Fulah": "ff",
+            "Finnish": "fi",
+            "French": "fr",
+            "Western Frisian": "fy",
+            "Irish": "ga",
+            "Gaelic": "gd",
+            "Galician": "gl",
+            "Gujarati": "gu",
+            "Hausa": "ha",
+            "Hebrew": "he",
+            "Hindi": "hi",
+            "Croatian": "hr",
+            "Haitian": "ht",
+            "Hungarian": "hu",
+            "Armenian": "hy",
+            "Indonesian": "id",
+            "Igbo": "ig",
+            "Iloko": "ilo",
+            "Icelandic": "is",
+            "Italian": "it",
+            "Japanese": "ja",
+            "Javanese": "jv",
+            "Georgian": "ka",
+            "Kazakh": "kk",
+            "Central Khmer": "km",
+            "Kannada": "kn",
+            "Korean": "ko",
+            "Luxembourgish": "lb",
+            "Ganda": "lg",
+            "Lingala": "ln",
+            "Lao": "lo",
+            "Lithuanian": "lt",
+            "Latvian": "lv",
+            "Malagasy": "mg",
+            "Macedonian": "mk",
+            "Malayalam": "ml",
+            "Mongolian": "mn",
+            "Marathi": "mr",
+            "Malay": "ms",
+            "Burmese": "my",
+            "Nepali": "ne",
+            "Dutch": "nl",
+            "Norwegian": "no",
+            "Northern Sotho": "ns",
+            "Occitan": "oc",
+            "Oriya": "or",
+            "Panjabi": "pa",
+            "Polish": "pl",
+            "Pushto": "ps",
+            "Portuguese": "pt",
+            "Romanian": "ro",
+            "Russian": "ru",
+            "Sindhi": "sd",
+            "Sinhala": "si",
+            "Slovak": "sk",
+            "Slovenian": "sl",
+            "Somali": "so",
+            "Albanian": "sq",
+            "Serbian": "sr",
+            "Swati": "ss",
+            "Sundanese": "su",
+            "Swedish": "sv",
+            "Swahili": "sw",
+            "Tamil": "ta",
+            "Thai": "th",
+            "Tagalog": "tl",
+            "Tswana": "tn",
+            "Turkish": "tr",
+            "Ukrainian": "uk",
+            "Urdu": "ur",
+            "Uzbek": "uz",
+            "Vietnamese": "vi",
+            "Wolof": "wo",
+            "Xhosa": "xh",
+            "Yiddish": "yi",
+            "Yoruba": "yo",
+            "Chinese": "zh",
+            "Zulu": "zu",
+        }
+        
+        
+        # Text input
+        user_input = st.text_area("Enter your text", " Hi Mohammed, it is nice to meet you. Let's discuss how to be better friends and tackle the world's issues such as global warming and climate change together.  There are already so many technology solutions in this grand city that can be applied to the world.  I am glad to be working on this problem with you.")
+        
+        # Language selection - for demonstration purposes only
+        # In a real translation scenario, you'd use actual language codes and a translation model
+        
+        source_lang = st.selectbox(label="Source language", options=list(lang_id.keys()))
+        target_lang = st.selectbox(label="Target language", options=list(lang_id.keys()))
+        
+        
+        # Button to generate text
+        if st.button("Generate"):
+        prompt = f"Translate the following {source_lang} text to {target_lang}: " + user_input + " ONLY TRANSLATE DON'T ADD ANY ADDITIONAL DETAILS"
+        
+        # Generate text
+        output = generate_text(prompt)
+        st.text_area("Generated Text", output, height=300)
+
+
         st.header("Custom GPT Engineering Tools")
         st.link_button("Conversation Analyzer", "https://chat.openai.com/g/g-XARuyBgpL-conversation-analyzer")