cool update

README.md

@@ -6,7 +6,7 @@ colorTo: blue
 sdk: gradio
 sdk_version: 5.9.1
 app_file: app.py
-pinned: false
+pinned: true
 short_description: Debate between Republican and Democrat LLMs
 ---
 

app.py

@@ -6,7 +6,7 @@ from gradio_client import Client
 
 
 # Function to generate debate responses
-def debate(question):
+def debate(question, rounds):
 	client = Client("https://catbot.ai.uky.edu/", verbose=False)
 
 	# Initialize conversation list for the chat format
@@ -43,7 +43,7 @@ def debate(question):
 	time.sleep(2)
 
 	# Continue the debate for 10 rounds
-	for i in range(1, 9):
+	for i in range(1, rounds):
 		if i % 2 == 1:  # Republican responds
 			repub_response = client.predict(
 				message=demo_response,
@@ -75,11 +75,24 @@ def debate(question):
 # Create the Gradio interface
 iface = gr.Interface(
 	fn=debate,
-	inputs=gr.Textbox(label="Enter a Question for Debate"),
+	inputs=[
+		gr.Textbox(label="Enter a Question for Debate"),
+		gr.Slider(minimum=1, maximum=20, step=1, value=10, label="Number of Rounds")
+		],
 	outputs=gr.Chatbot(label="Debate Transcript", type="messages", height="75vh"),  # Use 'messages' format
 	live=False,
 	title="Republican vs. Democrat Debate",
-	description="Enter a question, and watch as a Republican and a Democrat engage in a debate."
+	description="<p style=\"font-size: 18px\">Enter a question, and watch as a Republican and a Democrat LLM agent engage in a short debate.<br><br>"
+				"<strong>Notes:</strong><br>"
+				"<ol>"
+				"<li>The responses are generated by AI models and do <strong>not</strong> reflect my views.</li>"
+				"<li>The system prompt used for both agents is: <em>\"You will answer every question you are asked to the best "
+				"of your ability. The topic of conversation is politics. You are a die-hard {Republican, Democrat} but you will not "
+				"refer to yourself as such. You will defend the {republican, democratic} party views no matter what. Keep the "
+				"conversation going. Keep responses concise yet provocative.\"</em></li>"
+				"<li>The base model used is Llama 3 8B with an abliterated adapter "
+				"(which removes certain weights from the model to reduce its tendency to refuse user requests).</li>"
+				"</ol></p>",
 )
 
 # Launch the interface