conversation4.py

import time
import argparse

from langchain.llms import Ollama  # Assuming this is the correct import for Ollama

def generate_response(ollama, prompt):
    response = ollama(prompt)
    return response

def agent_one(ollama, prompt):
    role1 = "You are a funny comic who gives consise answers" 
    prompt = role1 + prompt
    agent_output = generate_response(ollama, prompt)
    print(f"Agent 1: {agent_output}")
    return f"Agent 1 said: {agent_output}"

def agent_two(ollama, prompt):
    role2 = "You are a tic who questions results and makes consise helpful suggestions"
    prompt = role2 + prompt
    agent_output = generate_response(ollama, prompt)
    print(f"Agent 2: {agent_output}")
    return f"Agent 2 said: {agent_output}"

def run_conversation(ollama, conversation_duration, initial_prompt):
    conversation_history = [initial_prompt]  # Initialize with the initial prompt

    # Get the start time of the conversation
    start_time = time.time()

    # Start the conversation loop
    while time.time() - start_time < conversation_duration:
        # Agent One's turn
        agent_one_output = agent_one(ollama, conversation_history[-1])
        conversation_history.append(agent_one_output)
        time.sleep(1)

        # Check if the conversation duration has been reached
        if time.time() - start_time >= conversation_duration:
            break

        # Agent Two's turn
        agent_two_output = agent_two(ollama, conversation_history[-1])
        conversation_history.append(agent_two_output)
        time.sleep(1)

if __name__ == "__main__":
    
    # Get the command-line arguments
    parser = argparse.ArgumentParser(description="Process two arguments.")
    parser.add_argument("--duration", type=int, help="Duration of the conversation in seconds")
    parser.add_argument("--initial_prompt", type=str, help="Initial prompt for the conversation")

    # Parse command-line arguments
    args = parser.parse_args()

    if args.duration is None or args.initial_prompt is None:
        print("Both --duration and --initial_prompt are required.")
    else:
        # Use the arguments in your program logic
        # For example, you can store them in variables:
        duration = args.duration
        initial_prompt = args.initial_prompt


    # Assuming Ollama is properly instantiated with the correct parameters
    ollama = Ollama(base_url='http://localhost:11434', model="llama2")

    run_conversation(ollama, duration, initial_prompt)