File size: 3,502 Bytes
b228d02
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
import os
import gradio as gr
import requests
from typing import List, Dict, Tuple

# Define the API URL (adjust according to your server address)
API_URL = "http://127.0.0.1:5000/chat"

History = List[Tuple[str, str]]
Messages = List[Dict[str, str]]

def clear_session() -> History:
    return []

def history_to_messages(history: History) -> Messages:
    messages = []
    for h in history:
        messages.append({'role': 'user', 'content': h[0].strip()})
        messages.append({'role': 'assistant', 'content': h[1].strip()})
    return messages

def messages_to_history(messages: Messages) -> History:
    history = []
    for q, r in zip(messages[0::2], messages[1::2]):
        history.append((q['content'], r['content']))
    return history

def model_chat(query: str, history: History) -> Tuple[str, History]:
    if not query.strip():
        return '', history
    
    messages = history_to_messages(history)
    messages.append({'role': 'user', 'content': query.strip()})

    response = requests.post(API_URL, json={"messages": messages})
    if response.status_code != 200:
        return "Error: Failed to get response from the API", history
    
    response_json = response.json()
    response_text = response_json["response"]

    history.append((query.strip(), response_text.strip()))
    return response_text.strip(), history

with gr.Blocks(css='''
    .gr-group {direction: rtl;}
    .chatbot{text-align:right;}
    .dicta-header {
        background-color: var(--input-background-fill);
        border-radius: 10px;
        padding: 20px;
        text-align: center;
        display: flex;
        flex-direction: row;
        align-items: center;
        box-shadow: var(--block-shadow);
        border-color: var(--block-border-color);
        border-width: 1px;
    }
    @media (max-width: 768px) {
        .dicta-header {
            flex-direction: column;
        }
    }
    .chatbot.prose {
        font-size: 1.2em;
    }
    .dicta-logo {
        width: 150px;
        height: auto;
        margin-bottom: 20px;
    }
    .dicta-intro-text {
        margin-bottom: 20px;
        text-align: center;
        display: flex;
        flex-direction: column;
        align-items: center;
        width: 100%;
        font-size: 1.1em;
    }
    textarea {
        font-size: 1.2em;
    }
''', js=None) as demo:
    gr.Markdown("""
<div class="dicta-header">
  <a href="">
    <img src="\\logo111.png" alt="Logo" class="dicta-logo">
  </a>  
  <div class="dicta-intro-text">
    <h1>爪'讗讟 诪注专讻讬 - 讛讚讙诪讛 专讗砖讜谞讬转</h1>
    <span dir='rtl'>讘专讜讻讬诐 讛讘讗讬诐 诇讚诪讜 讛讗讬谞讟专讗拽讟讬讘讬 讛专讗砖讜谉. 讞拽专讜 讗转 讬讻讜诇讜转 讛诪讜讚诇 讜专讗讜 讻讬爪讚 讛讜讗 讬讻讜诇 诇住讬讬注 诇讻谉 讘诪砖讬诪讜转讬讻诐</span><br/>
    <span dir='rtl'>讛讚诪讜 谞讻转讘 注诇 讬讚讬 住专谉 专讜注讬 专转诐 转讜讱 砖讬诪讜砖 讘诪讜讚诇 砖驻讛 讚讬拽讟讛 砖驻讜转讞 注诇 讬讚讬 诪驻讗"转</span><br/>
  </div>
</div>
""")
    
    chatbot = gr.Chatbot(height=600)
    query = gr.Textbox(placeholder="讛讻谞住 砖讗诇讛 讘注讘专讬转 (讗讜 讘讗谞讙诇讬转!)", rtl=True)
    clear_btn = gr.Button("谞拽讛 砖讬讞讛")

    def respond(query, history):
        response, history = model_chat(query, history)
        return history, gr.update(value="", interactive=True)

    demo_state = gr.State([])

    query.submit(respond, [query, demo_state], [chatbot, query, demo_state])
    clear_btn.click(clear_session, [], demo_state, chatbot)

demo.launch(share=True)