Update simple_dialin.py

simple_dialin.py

@@ -7,61 +7,39 @@ import argparse
 import asyncio
 import os
 import sys
-
-from call_connection_manager import CallConfigManager, SessionManager
-from dotenv import load_dotenv
 from loguru import logger
 
-from pipecat.adapters.schemas.function_schema import FunctionSchema
-from pipecat.adapters.schemas.tools_schema import ToolsSchema
+from call_connection_manager import CallConfigManager
 from pipecat.audio.vad.silero import SileroVADAnalyzer
-from pipecat.frames.frames import EndTaskFrame
+from pipecat.frames.frames import EndTaskFrame, LLMMessagesFrame
 from pipecat.pipeline.pipeline import Pipeline
 from pipecat.pipeline.runner import PipelineRunner
 from pipecat.pipeline.task import PipelineParams, PipelineTask
 from pipecat.processors.aggregators.openai_llm_context import OpenAILLMContext
-from pipecat.processors.frame_processor import FrameDirection
 from pipecat.services.cartesia.tts import CartesiaTTSService
-from pipecat.services.llm_service import FunctionCallParams
 from pipecat.services.openai.llm import OpenAILLMService
 from pipecat.transports.services.daily import DailyDialinSettings, DailyParams, DailyTransport
 
-load_dotenv(override=True)
-
 logger.remove(0)
 logger.add(sys.stderr, level="DEBUG")
 
-daily_api_key = os.getenv("DAILY_API_KEY", "")
-daily_api_url = os.getenv("DAILY_API_URL", "https://api.daily.co/v1")
-
-
-async def main(
-    room_url: str,
-    token: str,
-    body: dict,
-):
-    # ------------ CONFIGURATION AND SETUP ------------
-
-    # Create a config manager using the provided body
+async def main(room_url: str, token: str, body: dict):
     call_config_manager = CallConfigManager.from_json_string(body) if body else CallConfigManager()
-
-    # Get important configuration values
+    caller_info = call_config_manager.get_caller_info()
+    caller_number = caller_info["caller_number"]
+    dialed_number = caller_info["dialed_number"]
     test_mode = call_config_manager.is_test_mode()
-
-    # Get dialin settings if present
     dialin_settings = call_config_manager.get_dialin_settings()
 
-    # Initialize the session manager
-    session_manager = SessionManager()
+    logger.info(f"Caller number: {caller_number}")
+    logger.info(f"Dialed number: {dialed_number}")
 
     # ------------ TRANSPORT SETUP ------------
-
-    # Set up transport parameters
     if test_mode:
         logger.info("Running in test mode")
         transport_params = DailyParams(
-            api_url=daily_api_url,
-            api_key=daily_api_key,
+            api_url=os.environ.get("DAILY_API_URL", "https://api.daily.co/v1"),
+            api_key=os.environ.get("HF_DAILY_API_KEY", ""),
             audio_in_enabled=True,
             audio_out_enabled=True,
             video_out_enabled=False,
@@ -73,8 +51,8 @@ async def main(
             call_id=dialin_settings.get("call_id"), call_domain=dialin_settings.get("call_domain")
         )
         transport_params = DailyParams(
-            api_url=daily_api_url,
-            api_key=daily_api_key,
+            api_url=os.environ.get("DAILY_API_URL", "https://api.daily.co/v1"),
+            api_key=os.environ.get("HF_DAILY_API_KEY", ""),
             dialin_settings=daily_dialin_settings,
             audio_in_enabled=True,
             audio_out_enabled=True,
@@ -83,82 +61,33 @@ async def main(
             transcription_enabled=True,
         )
 
-    # Initialize transport with Daily
-    transport = DailyTransport(
-        room_url,
-        token,
-        "Simple Dial-in Bot",
-        transport_params,
-    )
-
-    # Initialize TTS
+    transport = DailyTransport(room_url, token, "Simple Dialin Bot", transport_params)
     tts = CartesiaTTSService(
-        api_key=os.getenv("CARTESIA_API_KEY", ""),
-        voice_id="b7d50908-b17c-442d-ad8d-810c63997ed9",  # Use Helpful Woman voice by default
-    )
-
-    # ------------ FUNCTION DEFINITIONS ------------
-
-    async def terminate_call(params: FunctionCallParams):
-        """Function the bot can call to terminate the call upon completion of a voicemail message."""
-        if session_manager:
-            # Mark that the call was terminated by the bot
-            session_manager.call_flow_state.set_call_terminated()
-
-        # Then end the call
-        await params.llm.queue_frame(EndTaskFrame(), FrameDirection.UPSTREAM)
-
-    # Define function schemas for tools
-    terminate_call_function = FunctionSchema(
-        name="terminate_call",
-        description="Call this function to terminate the call.",
-        properties={},
-        required=[],
+        api_key=os.environ.get("HF_CARTESIA_API_KEY", ""),
+        voice_id="b7d50908-b17c-442d-ad8d-810c63997ed9",
     )
-
-    # Create tools schema
-    tools = ToolsSchema(standard_tools=[terminate_call_function])
+    llm = OpenAILLMService(api_key=os.environ.get("HF_OPENAI_API_KEY"))
 
     # ------------ LLM AND CONTEXT SETUP ------------
-
-    # Set up the system instruction for the LLM
-    system_instruction = """You are Chatbot, a friendly, helpful robot. Your goal is to demonstrate your capabilities in a succinct way. Your output will be converted to audio so don't include special characters in your answers. Respond to what the user said in a creative and helpful way, but keep your responses brief. Start by introducing yourself. If the user ends the conversation, **IMMEDIATELY** call the `terminate_call` function. """
-
-    # Initialize LLM
-    llm = OpenAILLMService(api_key=os.getenv("OPENAI_API_KEY"))
-
-    # Register functions with the LLM
-    llm.register_function("terminate_call", terminate_call)
-
-    # Create system message and initialize messages list
+    system_instruction = """You are a friendly, helpful robot. Greet the user and ask how you can assist them."""
     messages = [call_config_manager.create_system_message(system_instruction)]
-
-    # Initialize LLM context and aggregator
-    context = OpenAILLMContext(messages, tools)
+    context = OpenAILLMContext(messages)
     context_aggregator = llm.create_context_aggregator(context)
 
     # ------------ PIPELINE SETUP ------------
+    pipeline = Pipeline([
+        transport.input(),
+        context_aggregator.user(),
+        llm,
+        tts,
+        transport.output(),
+        context_aggregator.assistant(),
+    ])
 
-    # Build pipeline
-    pipeline = Pipeline(
-        [
-            transport.input(),  # Transport user input
-            context_aggregator.user(),  # User responses
-            llm,  # LLM
-            tts,  # TTS
-            transport.output(),  # Transport bot output
-            context_aggregator.assistant(),  # Assistant spoken responses
-        ]
-    )
-
-    # Create pipeline task
     task = PipelineTask(pipeline, params=PipelineParams(allow_interruptions=True))
 
-    # ------------ EVENT HANDLERS ------------
-
     @transport.event_handler("on_first_participant_joined")
     async def on_first_participant_joined(transport, participant):
-        logger.debug(f"First participant joined: {participant['id']}")
         await transport.capture_participant_transcription(participant["id"])
         await task.queue_frames([context_aggregator.user().get_context_frame()])
 
@@ -168,25 +97,16 @@ async def main(
         await task.cancel()
 
     # ------------ RUN PIPELINE ------------
-
-    if test_mode:
-        logger.debug("Running in test mode (can be tested in Daily Prebuilt)")
-
     runner = PipelineRunner()
     await runner.run(task)
 
-
 if __name__ == "__main__":
-    parser = argparse.ArgumentParser(description="Simple Dial-in Bot")
+    parser = argparse.ArgumentParser(description="Pipecat Simple Dialin Bot")
     parser.add_argument("-u", "--url", type=str, help="Room URL")
     parser.add_argument("-t", "--token", type=str, help="Room Token")
     parser.add_argument("-b", "--body", type=str, help="JSON configuration string")
-
     args = parser.parse_args()
-
-    # Log the arguments for debugging
     logger.info(f"Room URL: {args.url}")
     logger.info(f"Token: {args.token}")
     logger.info(f"Body provided: {bool(args.body)}")
-
-    asyncio.run(main(args.url, args.token, args.body))
+    asyncio.run(main(args.url, args.token, args.body))