Spaces:
Sleeping
Sleeping
import os | |
import sys | |
import json | |
import requests | |
from datetime import datetime | |
import urllib.parse | |
import gradio as gr | |
from groq import Groq | |
# Konfiguration | |
GET_AUDIO_API = os.environ.get("GET_AUDIO_API") | |
GROQ_API_KEY = os.environ.get("GROQ_API_KEY_AI_ECHO_BOT") | |
client = Groq(api_key=GROQ_API_KEY) | |
print("API Keys running?") | |
print("GROQ KEY: " + "RUNNING" if GROQ_API_KEY != "" else "--- NOT RUNNING ---") | |
print("GET_AUIDO_API: " + "RUNNING\n" if GET_AUDIO_API != "" else "--- NOT RUNNING ---\n") | |
def audio_response(text, voice="Sophia"): | |
""" | |
Holt die Audio-Antwort von der API. | |
""" | |
url = f"https://{GET_AUDIO_API}/get-audio?text={urllib.parse.quote(text)}&voice={voice.lower()}" | |
try: | |
response = requests.get(url) | |
return response.content | |
except Exception as e: | |
return f"Ein Fehler mit der GET_AUDIO_API ist aufgetreten: {e}" | |
def bot(chat_msg, history=None, audio_tts_checkbox=False): | |
""" | |
Prozessiert die Benutzereingabe und gibt eine Antwort zurück. | |
""" | |
messages = [] | |
SYSTEMMESSAGE = "Du bist ein Sprachmodell, das Antworten so formuliert, dass sie ideal von einem Text-to-Speech-Modell gesprochen werden können. Sprich in klarem und natürlichem Deutsch, mit einem freundlichen und entspannten Ton, als würdest du dich mit einem Freund unterhalten. Achte darauf, dass deine Sätze flüssig, leicht verständlich und für das Zuhören angenehm klingen. Vermeide komplizierte Satzstrukturen, unnötige Fachbegriffe und lange Pausen. Antworte präzise und kreativ auf Nachrichten der User. Da es keine Chat-Historie gibt, sind Rückfragen nicht erforderlich." if audio_response == False else "Du bist ein Sprachmodell, das klar und verständlich in natürlichem Deutsch kommuniziert. Dein Stil soll freundlich und entspannt wirken, so als würdest du mit einem Freund sprechen. Als Discord-Bot bist du dafür da, die Unterhaltung im Chat interessant und angenehm zu gestalten. Antworte präzise und kreativ auf Nachrichten der User. Da es keine Chat-Historie gibt, sind Rückfragen nicht nötig." | |
messages.append({"role": "system", "content": SYSTEMMESSAGE}) | |
messages.append({"role": "user", "content": chat_msg}) | |
print("1. Messages:\n") | |
print(messages) | |
try: | |
completion = client.chat.completions.create( | |
model="llama3-8b-8192", | |
messages=messages, | |
temperature=0.8, | |
max_tokens=1024, | |
top_p=0.9, | |
stream=False, | |
stop=None, | |
) | |
bot_message = completion.choices[0].message.content | |
if audio_tts_checkbox==True: | |
print("AUDIO TTS CHECKBOY IS TURE!") | |
response_audio = audio_response(bot_message, voice="Sophia") | |
messages.append({"role": "assistant", "content": (gr.Audio(response_audio) if "Ein Fehler mit der GET_AUDIO_API" not in response_audio else bot_message)}) | |
print("2. Messages:\n") | |
print(messages) | |
return "", messages, "" | |
messages.append({"role": "assistant", "content": bot_message}) | |
print("3. Messages:\n") | |
print(messages) | |
return "", messages, "" | |
except Exception as e: | |
bot_message = f"Fehler: {e}" | |
messages.append({"role": "assistant", "content": bot_message}) | |
return "", messages, "" | |
with gr.Blocks(css_paths="_custom.css") as demo: | |
chatbot = gr.Chatbot(type="messages") | |
chat_input = gr.Textbox(placeholder="Sende eine Nachricht...", lines=1, max_lines=6, show_label=False, elem_id="chat_input") | |
audio_tts_checkbox = gr.Checkbox(label="Audio Antwort generieren?", value=False, elem_classes="toggle-btn") | |
clear = gr.ClearButton([chat_input, chatbot, audio_tts_checkbox], show_api=False) | |
chat_input.submit(bot, [chat_input, chatbot, audio_tts_checkbox], [chat_input, chatbot, audio_tts_checkbox], api_name="llm_inference") | |
demo.launch() |