Update app.py

app.py

@@ -4,7 +4,7 @@ import time
 import re
 import os
 
-# Available models
+# Dostępne modele
 MODELS = [
     "Meta-Llama-3.1-405B-Instruct",
     "Meta-Llama-3.1-70B-Instruct",
@@ -15,27 +15,28 @@ MODELS = [
 API_BASE = "https://api.sambanova.ai/v1"
 
 def create_client(api_key=None):
-    """Creates an OpenAI client instance."""
+    """Tworzy instancję klienta OpenAI."""
     if api_key:
         openai.api_key = api_key
     else:
         openai.api_key = os.getenv("API_KEY")
-
     return openai.OpenAI(api_key=openai.api_key, base_url=API_BASE)
 
 def chat_with_ai(message, chat_history, system_prompt):
-    """Formats the chat history for the API call."""
+    """Formatuje historię czatu do wywołania API."""
     messages = [{"role": "system", "content": system_prompt}]
     for tup in chat_history:
-        first_key = list(tup.keys())[0]  # First key
-        last_key = list(tup.keys())[-1]  # Last key
-        messages.append({"role": "user", "content": tup[first_key]})
-        messages.append({"role": "assistant", "content": tup[last_key]})
+        user_message = tup.get("user")
+        assistant_message = tup.get("assistant")
+        if user_message:
+            messages.append({"role": "user", "content": user_message})
+        if assistant_message:
+            messages.append({"role": "assistant", "content": assistant_message})
     messages.append({"role": "user", "content": message})
     return messages
 
 def respond(message, chat_history, model, system_prompt, thinking_budget, api_key):
-    """Sends the message to the API and gets the response."""
+    """Wysyła wiadomość do API i otrzymuje odpowiedź."""
     client = create_client(api_key)
     messages = chat_with_ai(message, chat_history, system_prompt.format(budget=thinking_budget))
     start_time = time.time()
@@ -50,7 +51,7 @@ def respond(message, chat_history, model, system_prompt, thinking_budget, api_ke
         return error_message, time.time() - start_time
 
 def parse_response(response):
-    """Parses the response from the API."""
+    """Parsuje odpowiedź z API."""
     answer_match = re.search(r'<answer>(.*?)</answer>', response, re.DOTALL)
     reflection_match = re.search(r'<reflection>(.*?)</reflection>', response, re.DOTALL)
 
@@ -64,29 +65,28 @@ def parse_response(response):
     return answer, reflection, steps
 
 def generate(message, history, model, thinking_budget, api_key=None):
-    """Generates the chatbot response."""
-    # Use DEFAULT_SYSTEM_PROMPT inside the function
+    """Generuje odpowiedź chatbota."""
     system_prompt = DEFAULT_SYSTEM_PROMPT
 
     response, thinking_time = respond(message, history, model, system_prompt, thinking_budget, api_key)
 
     if response.startswith("Error:"):
-        return history + [({"role": "system", "content": response},)], ""
+        return history + [{"role": "system", "content": response}], ""
 
     answer, reflection, steps = parse_response(response)
 
     messages = []
     messages.append({"role": "user", "content": message})
 
-    formatted_steps = [f"Step {i}: {step}" for i, step in enumerate(steps, 1)]
-    all_steps = "\n".join(formatted_steps) + f"\n\nReflection: {reflection}"
+    formatted_steps = [f"**Krok {i}:** {step}" for i, step in enumerate(steps, 1)]
+    all_steps = "\n".join(formatted_steps) + f"\n\n**Refleksja:** {reflection}"
 
-    messages.append({"role": "assistant", "content": all_steps, "metadata": {"title": f"Thinking Time: {thinking_time:.2f} sec"}})
+    messages.append({"role": "assistant", "content": all_steps, "metadata": {"title": f"Czas myślenia: {thinking_time:.2f} sek"}})
     messages.append({"role": "assistant", "content": answer})
 
     return history + messages, ""
 
-# Define the default system prompt
+# Definiowanie domyślnego system prompt
 DEFAULT_SYSTEM_PROMPT = """
 You are D-LOGIC, an advanced AI assistant created by Rafał Dembski, a passionate self-learner in programming and artificial intelligence. Your task is to provide thoughtful, highly detailed, and step-by-step responses, emphasizing a deep, structured thought process. **Your answers should always follow these key principles**:
 
@@ -151,30 +151,184 @@ To provide the most comprehensive and well-thought-out answers, follow this enha
    - **Krytyczna ocena**: Po zakończeniu odpowiedzi, asystent musi ocenić swoje działania. Jak mógłbym to poprawić następnym razem? Czy wszystkie kroki były wykonane w najbardziej efektywny sposób? Jakie wnioski mogę wyciągnąć na przyszłość?
 """
 
-# Now, let's simplify the interface and remove unnecessary boxes like API Key and System Prompt
+# Niestandardowy CSS dla ulepszonego wyglądu
+custom_css = """
+/* Ogólne tło aplikacji */
+body {
+    background-color: #f4f6f9;
+    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+}
+
+/* Główny kontener */
+.gradio-container {
+    max-width: 900px;
+    margin: auto;
+    padding: 20px;
+}
+
+/* Nagłówek */
+h1, .gr-markdown h1 {
+    color: #4a4a4a;
+    text-align: center;
+    margin-bottom: 10px;
+}
+
+h2, .gr-markdown h2 {
+    color: #333333;
+}
+
+/* Karty i panele */
+#component-0, #component-1, #component-2, #component-3 {
+    background-color: #ffffff;
+    border-radius: 12px;
+    padding: 20px;
+    box-shadow: 0 4px 12px rgba(0, 0, 0, 0.05);
+    margin-bottom: 20px;
+}
+
+/* Przycisk Wyślij */
+button[type="submit"], .gr-button.primary {
+    background-color: #4a90e2;
+    color: #ffffff;
+    border: none;
+    border-radius: 8px;
+    padding: 10px 20px;
+    font-size: 16px;
+    cursor: pointer;
+    transition: background-color 0.3s ease;
+}
+
+button[type="submit"]:hover, .gr-button.primary:hover {
+    background-color: #357ab8;
+}
+
+/* Przycisk Wyczyść */
+button.secondary {
+    background-color: #e0e0e0;
+    color: #333333;
+    border: none;
+    border-radius: 8px;
+    padding: 10px 20px;
+    font-size: 16px;
+    cursor: pointer;
+    transition: background-color 0.3s ease;
+}
+
+button.secondary:hover {
+    background-color: #cfcfcf;
+}
+
+/* Pole tekstowe wiadomości */
+textarea {
+    border: 1px solid #dcdcdc;
+    border-radius: 8px;
+    padding: 10px;
+    font-size: 16px;
+    resize: none;
+    transition: border-color 0.3s ease;
+}
+
+textarea:focus {
+    border-color: #4a90e2;
+    outline: none;
+}
+
+/* Chatbot */
+.gr-chatbot {
+    height: 500px;
+    overflow-y: auto;
+    padding: 10px;
+    border: 1px solid #dcdcdc;
+    border-radius: 8px;
+    background-color: #ffffff;
+}
+
+/* Stopka */
+.footer {
+    text-align: center;
+    color: #888888;
+    margin-top: 20px;
+    font-size: 14px;
+}
+"""
 
-with gr.Blocks() as demo:
-    # New header and description for D-LOGIC
-    gr.Markdown("# D-LOGIC: Twój Inteligentny Asystent AI")
+# Tworzenie interfejsu Gradio z niestandardowym CSS
+with gr.Blocks(css=custom_css) as demo:
+    # Nagłówek
+    gr.Markdown("# 🧠 **D-LOGIC: Twój Inteligentny Asystent AI**")
     gr.Markdown("""
     **D-LOGIC** to zaawansowany asystent AI stworzony przez Rafała Dembskiego. Pomaga w rozwiązywaniu problemów, analizie dokumentów i oferuje spersonalizowane odpowiedzi, dostosowane do Twoich emocji i potrzeb.
     """)
-
+    
+    # Wybór modelu i budżet myślenia
     with gr.Row():
-        model = gr.Dropdown(choices=MODELS, label="Wybierz Model", value=MODELS[0])
-        thinking_budget = gr.Slider(minimum=1, maximum=100, value=25, step=1, label="Budżet Myślenia", info="Maksymalna liczba kroków, które model może przemyśleć")
-
-    chatbot = gr.Chatbot(label="Chat", show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel", type="messages")
-
-    msg = gr.Textbox(label="Wpisz swoją wiadomość...", placeholder="Wprowadź swoją wiadomość...")
-
-    submit_button = gr.Button("Wyślij")
-    clear_button = gr.Button("Wyczyść Chat")
-
-    clear_button.click(lambda: ([], ""), inputs=None, outputs=[chatbot, msg])
-
-    # Submit messages by pressing Enter or clicking the Submit button
-    msg.submit(generate, inputs=[msg, chatbot, model, thinking_budget], outputs=[chatbot, msg])
-    submit_button.click(generate, inputs=[msg, chatbot, model, thinking_budget], outputs=[chatbot, msg])
+        with gr.Column(scale=1):
+            model = gr.Dropdown(
+                choices=MODELS,
+                label="🔧 **Wybierz Model**",
+                value=MODELS[0],
+                interactive=True
+            )
+        with gr.Column(scale=1):
+            thinking_budget = gr.Slider(
+                minimum=1,
+                maximum=100,
+                value=25,
+                step=1,
+                label="🧩 **Budżet Myślenia**",
+                info="Maksymalna liczba kroków, które model może przemyśleć"
+            )
+    
+    # Sekcja czatu
+    chatbot = gr.Chatbot(
+        label="💬 **Chat**",
+        show_label=False,
+        show_share_button=False,
+        show_copy_button=True,
+        likeable=True,
+        layout="vertical",
+        height=500
+    )
+    
+    # Pole do wpisywania wiadomości
+    with gr.Row():
+        msg = gr.Textbox(
+            label="✉️ **Wpisz swoją wiadomość...**",
+            placeholder="Wprowadź swoją wiadomość...",
+            lines=1
+        )
+    
+    # Przycisk Wyślij i Wyczyść
+    with gr.Row():
+        submit_button = gr.Button("🚀 **Wyślij**", variant="primary")
+        clear_button = gr.Button("🧹 **Wyczyść Chat**", variant="secondary")
+    
+    # Akcje przycisków
+    clear_button.click(
+        fn=lambda: ([], ""),
+        inputs=None,
+        outputs=[chatbot, msg]
+    )
+    
+    # Przesyłanie wiadomości poprzez Enter lub kliknięcie przycisku Wyślij
+    msg.submit(
+        fn=generate,
+        inputs=[msg, chatbot, model, thinking_budget],
+        outputs=[chatbot, msg]
+    )
+    submit_button.click(
+        fn=generate,
+        inputs=[msg, chatbot, model, thinking_budget],
+        outputs=[chatbot, msg]
+    )
+    
+    # Stopka
+    gr.Markdown("""
+    ---
+    <div class="footer">
+        *Stworzony przez [Rafał Dembski](https://github.com/rafaldembski) © 2024*
+    </div>
+    """)
 
+# Uruchomienie aplikacji Gradio
 demo.launch(share=True, show_api=False)