Update app.py

app.py

@@ -2,32 +2,39 @@ import os
 import gradio as gr
 from litellm import completion
 
-# встановлюємо змінну середовища для COHERE_API_KEY; мій trial ключ
-os.environ["COHERE_API_KEY"] = 'AbTU8gQ2cjzBDseA5SB6umXMGFFH3azjKQBQ6iq3'
-
 
 # Функція для отримання відповіді від Cohere
-def get_cohere_response(user_input):
+def get_cohere_response(user_input, api_key):
+    # Перевірка наявності ключа
+    if not api_key:
+        return "Схоже, ви не ввели ключ авторизації. Введіть його та спробуйте ще раз."
+
+    # Встановлюємо ключ API
+    os.environ["COHERE_API_KEY"] = api_key
 
-    # перевіримо довжину запиту
-    if len(user_input) < 5:
-        return "Схоже, щось пішло не так. Спробуй перефразувати своє питання у форматі '*поняття* це?' " \
-               "або 'що таке *поняття*?' та надішли запит ще раз.\n Looks like something went wrong. Use the " \
-               "formulation '*concept* is?' or 'what is *concept*?' and try again."
+    try:
+        # Перевіримо довжину запиту
+        if len(user_input) < 5:
+            return "Схоже, щось пішло не так. Спробуй перефразувати своє питання у форматі '*поняття* це?' " \
+                   "або 'що таке *поняття*?' та надішли запит ще раз.\n Looks like something went wrong. Use the " \
+                   "formulation '*concept* is?' or 'what is *concept*?' and try again."
 
-    # вказівка щодо формату відповіді
-    prompt = f"In other cases explain the question in simple words. Give some analogy from real life. " \
-             f"If the text is not in English, write the answer using language of the question. " \
-             f"Question: {user_input}"
+        # Вказівка щодо формату відповіді
+        prompt = f"Explain the question in simple words. Give some analogy from real life. " \
+                 f"If the text is not in English, write the answer using language of the question. " \
+                 f"Question: {user_input}"
 
-    response = completion(
-        model="command-r",  # Модель "command-r"
-        messages=[{
-            "content": prompt,
-            "role": "user"
-        }]
-    )
-    return response.choices[0].message.content  # лише відповідь-текст
+        # Викликаємо API
+        response = completion(
+            model="command-r",  # Модель "command-r"
+            messages=[{
+                "content": prompt,
+                "role": "user"
+            }]
+        )
+        return response.choices[0].message.content  # лише відповідь-текст
+    except Exception as e:
+        return f"Схоже, щось пішло не так: {str(e)}. Перевірте ваш ключ і спробуйте ще раз."
 
 
 # Створюємо інтерфейс Gradio
@@ -38,6 +45,11 @@ def interface():
                     " можливість нарешті розшифрувати їх на просту й доступну мову, доповнену аналогіями із реального "
                     "життя. Просто введіть запит, натисніть кнопку — і отримаєте відповідь! "
                     "<br> _*З використанням моделі [Cohere Command-R](https://cohere.com)._")
+
+        # Поле для введення ключа
+        api_key = gr.Textbox(label="Введіть ваш Cohere API ключ", type="password",
+                             placeholder="Введіть ключ авторизації")
+
         # Вхідне текстове поле
         user_input = gr.Textbox(label="Введіть запит", placeholder="Що таке квантова механіка?")
 
@@ -48,9 +60,9 @@ def interface():
         submit_btn = gr.Button("Отримати відповідь")
 
         # Визначення дії при натисканні кнопки
-        submit_btn.click(get_cohere_response, inputs=user_input, outputs=output)
+        submit_btn.click(get_cohere_response, inputs=[user_input, api_key], outputs=output)
 
-    iface.launch()
+    iface.launch(share=True)  # Додаємо share=True для створення публічного посилання
 
 
-interface()  # виклик інтерфейсу
+interface()  # Виклик інтерфейсу