Spaces:
Runtime error
Runtime error
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)
|