Hugging Face's logo

Spaces:
nickyJames
/
Alfred_RAG
Sleeping

App Files Community
Alfred_RAG
File size: 4,930 Bytes
27e9eb3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
 
# app.py
# Alfred - The Gala Party Agent with Agentic RAG
# Unit 3 Use Case Project for Hugging Face Agents Course

import os
import gradio as gr
from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel

# Import our custom tools
from retriever import create_guest_retriever_tool
from tools import get_current_time, calculate_party_budget, suggest_seating_arrangement, dietary_check


# ============================================
# SETUP THE MODEL
# ============================================

model = InferenceClientModel(
    model_id="Qwen/Qwen2.5-7B-Instruct",  # Free tier compatible model
    token=os.environ.get("HF_TOKEN"),
)


# ============================================
# CREATE THE GUEST RETRIEVER TOOL (RAG)
# ============================================

print("Loading guest database...")
guest_retriever_tool = create_guest_retriever_tool()
print("Guest database loaded successfully!")


# ============================================
# CREATE ALFRED - THE GALA AGENT
# ============================================

alfred = CodeAgent(
    model=model,
    tools=[
        guest_retriever_tool,      # RAG tool for guest information
        DuckDuckGoSearchTool(),    # Web search for general queries
        get_current_time,          # Time tool
        calculate_party_budget,    # Budget calculator
        suggest_seating_arrangement,  # Seating planner
        dietary_check,             # Dietary requirements helper
    ],
    max_steps=5,
    verbosity_level=1,
)


# ============================================
# CHAT FUNCTION
# ============================================

def chat_with_alfred(message, history):
    """
    Process user message and return Alfred's response.
    """
    if not message.strip():
        return "Good day! I am Alfred, your gala assistant. How may I help you with the party preparations?"
    
    try:
        # Add context to help Alfred understand his role
        enhanced_message = f"""You are Alfred, the butler and gala party assistant. 
You have access to a guest database that you can search using the guest_info_retriever tool.
You can also help with party planning, budgets, seating, and dietary requirements.

User's request: {message}

If the question is about a guest or guests, use the guest_info_retriever tool to find information.
Be helpful, polite, and thorough in your responses."""
        
        response = alfred.run(enhanced_message)
        return str(response)
    
    except Exception as e:
        error_msg = str(e)
        if "404" in error_msg or "not found" in error_msg.lower():
            return "I apologize, but I'm having trouble connecting to my systems. Please ensure the HF_TOKEN is properly configured."
        elif "rate" in error_msg.lower() or "limit" in error_msg.lower():
            return "I've reached my service limits momentarily. Please wait a moment and try again."
        else:
            return f"I encountered an unexpected issue: {error_msg}"


# ============================================
# GRADIO INTERFACE
# ============================================

with gr.Blocks(title="Alfred - Gala Party Agent") as demo:
    
    gr.Markdown(
        """
        # 🎩 Alfred - Your Gala Party Assistant
        
        Good evening! I am Alfred, your dedicated assistant for the grand gala. 
        I have access to the complete guest list and can help you with all party preparations.
        
        ### What I Can Do:
        - 🔍 **Guest Information** - Search and retrieve details about any guest
        - 🌐 **Web Search** - Find information from the internet
        - 🕐 **Current Time** - Tell you the current date and time
        - 💰 **Budget Calculator** - Calculate party expenses
        - 🪑 **Seating Planner** - Suggest seating arrangements
        - 🥗 **Dietary Helper** - Get menu suggestions for dietary requirements
        
        ---
        """
    )
    
    chatbot = gr.ChatInterface(
        fn=chat_with_alfred,
        examples=[
            "Who is Lady Galadriel and what are her interests?",
            "Which guests are vegetarian?",
            "Tell me about guests who like music",
            "Calculate budget for 50 guests at $100 each, $5000 venue, $2000 entertainment",
            "What's the current time?",
            "Suggest seating for: John, Mary, Bob, Alice, Tom, Jane at tables of 3",
            "What are vegan menu options?",
        ],
    )
    
    gr.Markdown(
        """
        ---
        *🎓 Unit 3 Agentic RAG Use Case | Hugging Face Agents Course*
        
        This agent demonstrates **Agentic RAG** - combining an AI agent with document retrieval 
        to answer questions about specific data (guest information).
        """
    )


# ============================================
# LAUNCH
# ============================================

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