Update app.py

app.py

@@ -1,33 +1,30 @@
-# app.py
-
+#### Конечный:
 import os
-from http.client import HTTPMessage
-
-os.system('pip install dashscope')
-
-import gradio as gr
 from http import HTTPStatus
-import dashscope
-from dashscope import Generation
-from dashscope.api_entities.dashscope_response import Role
+import gradio as gr
+from dashscope import Generation, Role
 from typing import List, Optional, Tuple, Dict
 from urllib.error import HTTPError
+import unittest
 
 default_system = 'You are Qwen, created by Alibaba Cloud. You are a helpful assistant.'
 
-YOUR_API_TOKEN = os.getenv('YOUR_API_TOKEN')
-dashscope.api_key = YOUR_API_TOKEN
+GROQ_API_KEY = os.environ.get('GROQ_API_KEY')
+client = OpenAI(
+    api_key=GROQ_API_KEY,
+    base_url="https://api.groq.com/openai/v1",
+)
 
 History = List[Tuple[str, str]]
 Messages = List[Dict[str, str]]
 
 
-def clear_session() -> History:
+def clear_session() -> Tuple[str, History]:
     return '', []
 
 
-def modify_system_session(system: str) -> str:
-    if system is None or len(system) == 0:
+def modify_system_session(system: str) -> Tuple[str, str, History]:
+    if not system:
         system = default_system
     return system, system, []
 
@@ -49,8 +46,7 @@ def messages_to_history(messages: Messages) -> Tuple[str, History]:
     return system, history
 
 
-def model_chat(query: Optional[str], history: Optional[History], system: str, radio: str
-               ) -> Tuple[str, str, History]:
+def model_chat(query: Optional[str], history: Optional[History], system: str, radio: str) -> Tuple[str, str, History]:
     if query is None:
         query = ''
     if history is None:
@@ -58,51 +54,51 @@ def model_chat(query: Optional[str], history: Optional[History], system: str, ra
     messages = history_to_messages(history, system)
     messages.append({'role': Role.USER, 'content': query})
     label_model = f"qwen2.5-coder-{radio.lower()}-instruct"
-    gen = Generation.call(
-        model=label_model,
-        messages=messages,
-        result_format='message',
-        stream=True
-    )
-    for response in gen:
-        if response.status_code == HTTPStatus.OK:
-            role = response.output.choices[0].message.role
-            response = response.output.choices[0].message.content
-            system, history = messages_to_history(messages + [{'role': role, 'content': response}])
-            yield '', history, system
-        else:
-            raise HTTPError(code=404, msg='Request id: %s, Status code: %s, error code: %s, error message: %s' % (
-                response.request_id, response.status_code,
-                response.code, response.message), hdrs=HTTPMessage(), url='http://example.com', fp=None)
-
-
-def chiose_radio(radio, system):
+    try:
+        gen = Generation.call(
+            model=label_model,
+            messages=messages,
+            result_format='message',
+            stream=True
+        )
+        for response in gen:
+            if response.status_code == HTTPStatus.OK:
+                role = response.output.choices[0].message.role
+                response = response.output.choices[0].message.content
+                system, history = messages_to_history(messages + [{'role': role, 'content': response}])
+                yield '', history, system
+            else:
+                raise HTTPError(code=response.status_code, msg='Request failed with status code: %s' % response.status_code)
+    except HTTPError as e:
+        print(f"HTTP error occurred: {e}")
+    except Exception as e:
+        print(f"An error occurred: {e}")
+
+
+def choose_radio(radio, system):
     mark_ = gr.Markdown(value=f"<center><font size=8>Qwen2.5-Coder-{radio}-instruct👾</center>")
     chatbot = gr.Chatbot(label=f'Qwen2.5-Coder-{radio.lower()}-instruct')
     
-    if system is None or len(system) == 0:
+    if not system:
         system = default_system
     
     return mark_, chatbot, system, system, ""
 
 
 def update_other_radios(value, other_radio1, other_radio2):
-    if value == "":
-        if other_radio1 != "":
-            selected = other_radio1
-        else:
-            selected = other_radio2
+    if not value:
+        selected = other_radio1 or other_radio2
         return selected, other_radio1, other_radio2
     return value, "", ""
 
 
 def main():
-    # 创建两个标签
+    # Создание интерфейса Gradio
     with gr.Blocks() as demo:
         with gr.Row():
             options_coder = ["0.5B", "1.5B", "3B", "7B", "14B", "32B", ]
             with gr.Row():
-                radio = gr.Radio(choices=options_coder, label="Qwen2.5-Coder：", value="32B")
+                radio = gr.Radio(choices=options_coder, label="Qwen2.5-Coder:", value="32B")
         
         with gr.Row():
             with gr.Accordion():
@@ -134,7 +130,7 @@ def main():
                                     inputs=[system_input],
                                     outputs=[system_state, system_input, chatbot])
         
-        radio.change(chiose_radio,
+        radio.change(choose_radio,
                      inputs=[radio, system_input],
                      outputs=[mark_, chatbot, system_state, system_input, textbox])
     
@@ -143,4 +139,22 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()
+
+# Тесты
+class TestApp(unittest.TestCase):
+    def test_clear_session(self):
+        self.assertEqual(clear_session(), ('', []))
+
+    def test_modify_system_session(self):
+        self.assertEqual(modify_system_session(None), (default_system, default_system, []))
+        self.assertEqual(modify_system_session(""), (default_system, default_system, []))
+        self.assertEqual(modify_system_session("Custom System"), ("Custom System", "Custom System", []))
+
+    def test_update_other_radios(self):
+        self.assertEqual(update_other_radios("", "32B", ""), ("32B", "32B", ""))
+        self.assertEqual(update_other_radios("", "", "14B"), ("14B", "", "14B"))
+        self.assertEqual(update_other_radios("7B", "", ""), ("7B", "", ""))
+
+if __name__ == "__main__":
+    unittest.main()