Update gigiachat_requests.py

gigiachat_requests.py

@@ -76,4 +76,47 @@ def get_completion_from_gigachat(prompt, max_tokens, access_token):
     response_data = response.json()
     answer_from_llm = response_data['choices'][0]['message']['content']
 
-    return answer_from_llm
+    return answer_from_llm
+
+def process_transcribation_with_gigachat(prompt, transcript, access_token):
+    url_completion = os.getenv('GIGA_COMPLETION_URL')
+
+    data_copm = json.dumps({
+        "model": 'GigaChat-Plus',
+        "messages": [
+        {
+            "role": "user",
+            "content": prompt + transcript
+        }
+        ],
+        "stream": True,
+    })
+        
+    headers_comp = {
+        'Content-Type': 'application/json',
+        'Accept': 'application/json',
+        'Authorization': 'Bearer ' + access_token
+    }
+
+    output_text = ''
+    st.write("Результат обработки:")
+    text_container = st.empty()
+    response = requests.post(url_completion, headers=headers_comp, data=data_copm, verify=False, stream=True)
+    for line in response.iter_lines():
+        if line:
+            decoded_line = line.decode('utf-8')
+            if decoded_line.startswith('data:'):
+                decoded_line = decoded_line[len('data:'):].strip()
+            try:
+                data = json.loads(decoded_line)
+                if "choices" in data:
+                    for choice in data["choices"]:
+                        if "delta" in choice and "content" in choice["delta"]:
+                            output_text += choice["delta"]["content"]
+                            text_container.text(output_text)
+            except json.JSONDecodeError:
+                continue
+
+
+    return output_text
+