Initial commit

Dockerfile

@@ -0,0 +1,28 @@
+FROM python:3.13.1
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
+
+# Switch to the "user" user
+USER user
+
+# Set home to the user's home directory
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
+
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+COPY --chown=user . $HOME/app
+
+# Expose port for local testing
+EXPOSE 7860
+
+CMD ["python", "main.py"]

main.py

@@ -0,0 +1,171 @@
+import gradio as gr
+from gradio import ChatMessage
+from agents import Agent, Runner, function_tool, FileSearchTool
+
+VECTOR_STORE_ID = "vs_6896d8c959008191981d645850b42313"
+
+
+@function_tool
+def get_bytes_cafe_menu(date: str) -> any:
+    return {
+        f"{date}": {
+            "daily byte": {
+                "name": "Steak Quesadilla",
+                "price": 12,
+                "description": "Flank steak, mixed cheese in a flour tortilla served with air fried potatoes, sour cream and salsa",
+            },
+            "vegetarian": {
+                "name": "Impossible Quesadilla",
+                "price": 12,
+                "description": "Impossible plant based product, mixed cheese in a flour tortilla served with air fried potatoes, sour cream and salsa",
+            },
+            "international": {
+                "name": "Chicken Curry",
+                "price": 12,
+                "description": "Chicken thighs, onion, carrot, potato, curry sauce served over rice",
+            },
+        }
+    }
+
+
+cafe_agent = Agent(
+    name="Cafe Agent",
+    instructions="You help students locate and provide information about the Bytes Cafe.",
+    tools=[
+        get_bytes_cafe_menu,
+    ],
+)
+
+
+building_agent = Agent(
+    name="Building Agent",
+    instructions="You help students locate and provide information about buildings and rooms on campus. Be descriptive when giving locations.",
+    tools=[
+        FileSearchTool(
+            max_num_results=3,
+            vector_store_ids=[VECTOR_STORE_ID],
+            include_search_results=True,
+        )
+    ],
+)
+
+course_agent = Agent(
+    name="Course Agent",
+    instructions="You help students find out information about courses held at DigiPen.",
+    tools=[
+        FileSearchTool(
+            max_num_results=5,
+            vector_store_ids=[VECTOR_STORE_ID],
+            include_search_results=True,
+        )
+    ],
+)
+
+handbook_agent = Agent(
+    name="Handbook Agent",
+    instructions="You help students navigate the school handbook, providing information about campus policies and student conduct.",
+    tools=[
+        FileSearchTool(
+            max_num_results=5,
+            vector_store_ids=[VECTOR_STORE_ID],
+            include_search_results=True,
+        )
+    ],
+)
+
+agent = Agent(
+    name="DigiPen Campus Assistant",
+    instructions="You are a helpful campus assistant that can plan and execute tasks for students at DigiPen. Please be concise and accurate in handing off tasks to other agents as needed.",
+    handoffs=[building_agent, course_agent, handbook_agent, cafe_agent],
+)
+
+
+async def chat_with_agent(user_msg: str, history: list):
+    messages = [{"role": msg["role"], "content": msg["content"]} for msg in history]
+    messages.append({"role": "user", "content": user_msg})
+    responses = []
+    reply_created = False
+    active_agent = None
+
+    result = Runner.run_streamed(agent, messages)
+    async for event in result.stream_events():
+        if event.type == "raw_response_event":
+            if event.data.type == "response.output_text.delta":
+                if not reply_created:
+                    responses.append(ChatMessage(role="assistant", content=""))
+                    reply_created = True
+                responses[-1].content += event.data.delta
+        elif event.type == "agent_updated_stream_event":
+            active_agent = event.new_agent.name
+            responses.append(
+                ChatMessage(
+                    content=event.new_agent.name,
+                    metadata={"title": "Agent Now Running", "id": active_agent},
+                )
+            )
+        elif event.type == "run_item_stream_event":
+            if event.item.type == "tool_call_item":
+                if event.item.raw_item.type == "file_search_call":
+                    responses.append(
+                        ChatMessage(
+                            content=f"Query used: {event.item.raw_item.queries}",
+                            metadata={
+                                "title": "File Search Completed",
+                                "parent_id": active_agent,
+                            },
+                        )
+                    )
+                else:
+                    tool_name = getattr(event.item.raw_item, "name", "unknown_tool")
+                    tool_args = getattr(event.item.raw_item, "arguments", {})
+                    responses.append(
+                        ChatMessage(
+                            content=f"Calling tool {tool_name} with arguments {tool_args}",
+                            metadata={"title": "Tool Call", "parent_id": active_agent},
+                        )
+                    )
+            if event.item.type == "tool_call_output_item":
+                responses.append(
+                    ChatMessage(
+                        content=f"Tool output: '{event.item.raw_item['output']}'",
+                        metadata={"title": "Tool Output", "parent_id": active_agent},
+                    )
+                )
+            if event.item.type == "handoff_call_item":
+                responses.append(
+                    ChatMessage(
+                        content=f"Name: {event.item.raw_item.name}",
+                        metadata={
+                            "title": "Handing Off Request",
+                            "parent_id": active_agent,
+                        },
+                    )
+                )
+        yield responses
+
+
+demo = gr.ChatInterface(
+    chat_with_agent,
+    title="DigiPen Campus Assistant",
+    theme=gr.themes.Soft(
+        primary_hue="red", secondary_hue="slate", font=[gr.themes.GoogleFont("Inter")]
+    ),
+    examples=[
+        "Where is the 'Hopper' room located?",
+        "I'm trying to find the WANIC classrooms. Can you help?",
+        "What's the policy for eating in auditoriums?",
+        "Where do I pickup my parking pass?",
+        "What's today's vegetarian dish at the Bytes Cafe?",
+        "Tell me more about CS 205...",
+        "What are the prerequisites for FLM201?",
+        "Which courses should I consider if I'm interested in audio mixing techniques?",
+    ],
+    submit_btn=True,
+    flagging_mode="manual",
+    flagging_options=["Like", "Spam", "Inappropriate", "Other"],
+    type="messages",
+    save_history=False,
+)
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)

requirements.txt

@@ -0,0 +1,2 @@
+gradio>=5.41.1
+openai-agents>=0.2.5