Hugging Face's logo

Spaces:
Remiscus
/
Customer_Support_Agent
Build error

App Files Community
Customer_Support_Agent
File size: 4,635 Bytes
6bc4708
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
114
115
116
117
118
119
120
121
122
123
124
125
126
import gradio as gr
import pandas as pd
import os
import atexit
from fastrtc import WebRTC, ReplyOnPause, get_stt_model, get_tts_model
from settings import Settings
from pydantic_ai.messages import (
    ModelMessage,
    ModelRequest,
    ModelResponse,
    UserPromptPart,
    TextPart
)
from agents import form_agent, response_agent

# Config and Globals
settings = Settings()
stt_model = get_stt_model()
tts_model = get_tts_model()
messages: list[ModelMessage] = []

DATA_PATH = "data.csv"
df = pd.read_csv(DATA_PATH) if os.path.exists(DATA_PATH) else pd.DataFrame(columns=["customer_name", "request_type", "issue", "emotion"])

def save_data_on_exit():
    df.to_csv(DATA_PATH, index=False)

atexit.register(save_data_on_exit)

def df_update():
    global df
    try:
        form_response = form_agent.run_sync(user_prompt="Do your thing", message_history=messages)
        new_row = {
            "customer_name": form_response.data.customername,
            "request_type": form_response.data.requesttype,
            "issue": form_response.data.issue,
            "emotion": form_response.data.emotion
        }
        df = pd.concat([df, pd.DataFrame([new_row])], ignore_index=True)
        df.to_csv(DATA_PATH, index=False)
        return "✅ DataFrame updated successfully!"
    except Exception as e:
        return f"❌ Update failed: {str(e)}"

def update_table():
    global df
    if os.path.exists(DATA_PATH):
        df = pd.read_csv(DATA_PATH)
    else:
        df = pd.DataFrame(columns=["customer_name", "request_type", "issue", "emotion"])
    return df

def reset_memory():
    global messages
    messages = []
    return "🧠 Memory reset successfully."

async def handle_audio(audio):
    prompt = stt_model.stt(audio)
    response_text = await response_agent.run(user_prompt=prompt, message_history=messages)
    messages.append(ModelRequest(parts=[UserPromptPart(content=prompt)]))
    messages.append(ModelResponse(parts=[TextPart(content=response_text.data)]))
    for chunk in tts_model.stream_tts(response_text.data):
        yield chunk

async def handle_text_chat(user_text, history):
    response = await response_agent.run(user_prompt=user_text, message_history=messages)
    messages.append(ModelRequest(parts=[UserPromptPart(content=user_text)]))
    messages.append(ModelResponse(parts=[TextPart(content=response.data)]))
    history = history + [[user_text, response.data]]
    return "", history

# Gradio UI
with gr.Blocks(css="""

.toolbox { display: flex; gap: 0.5rem; margin-top: 0.5rem; }

.footer-note { text-align: center; font-size: 0.85rem; color: #666; margin-top: 1rem; }

""") as demo:
    gr.Markdown("<h2 style='text-align: center;'>💬 Customer Support Assistant</h2>")

    debug_box = gr.Textbox(visible=False)

    with gr.Tabs():
        with gr.Tab("Chat"):
            with gr.Row():
                with gr.Column(scale=3):
                    chatbot = gr.Chatbot(label="Chat Interface")
                    user_input = gr.Textbox(placeholder="Type your message...", show_label=False)
                    user_input.submit(fn=handle_text_chat, inputs=[user_input, chatbot], outputs=[user_input, chatbot])

                with gr.Column(scale=1):
                    mic_button = WebRTC(mode="send-receive", modality="audio")
                    mic_button.stream(fn=ReplyOnPause(handle_audio), inputs=[mic_button], outputs=[mic_button], time_limit=60)

        with gr.Tab("Customer Data"):
            gr.Markdown("### Customer Information Table")
            data_frame = gr.Dataframe(
                headers=["customer_name", "request_type", "issue", "emotion"],
                interactive=False,
                wrap=True
            )
            with gr.Row(elem_classes="toolbox"):
                update_button = gr.Button("📤 Update DataFrame")
                refresh_button = gr.Button("🔄 Refresh Table")
                reset_button = gr.Button("🪹 Reset Memory")

            update_button.click(fn=df_update, outputs=[debug_box])
            refresh_button.click(fn=update_table, outputs=[data_frame])
            reset_button.click(fn=reset_memory, outputs=[debug_box])

    # Toast feedback
    def show_toast(msg: str):
        if msg:
            gr.Info(msg)

    debug_box.change(fn=show_toast, inputs=[debug_box])

    # Footer
    gr.Markdown("<div class='footer-note'>🚀 Made with ❤️ by Rikhil</div>")

    demo.load(fn=update_table, outputs=[data_frame])


if __name__ == "__main__":
    demo.launch()