Update app.py

app.py

@@ -81,28 +81,11 @@ def convert_to_xlam_tool(tools):
     else:
         return tools
 
-def build_conversation_history_prompt(conversation_history: str):
-    parsed_history = []
-    for step_data in conversation_history:
-        parsed_history.append({
-            "step_id": step_data["step_id"],
-            "thought": step_data["thought"],
-            "tool_calls": step_data["tool_calls"],
-            "next_observation": step_data["next_observation"],
-            "user_input": step_data['user_input']
-        })
-    
-    history_string = json.dumps(parsed_history)
-    return f"\n[BEGIN OF HISTORY STEPS]\n{history_string}\n[END OF HISTORY STEPS]\n"
-
-def build_prompt(task_instruction: str, format_instruction: str, tools: list, query: str, conversation_history: list):
+def build_prompt(task_instruction: str, format_instruction: str, tools: list, query: str):
     prompt = f"[BEGIN OF TASK INSTRUCTION]\n{task_instruction}\n[END OF TASK INSTRUCTION]\n\n"
     prompt += f"[BEGIN OF AVAILABLE TOOLS]\n{json.dumps(tools)}\n[END OF AVAILABLE TOOLS]\n\n"
     prompt += f"[BEGIN OF FORMAT INSTRUCTION]\n{format_instruction}\n[END OF FORMAT INSTRUCTION]\n\n"
     prompt += f"[BEGIN OF QUERY]\n{query}\n[END OF QUERY]\n\n"
-    
-    if conversation_history and len(conversation_history) > 0:
-        prompt += build_conversation_history_prompt(conversation_history)
     return prompt
 
 def generate_response(tools_input, query):
@@ -112,7 +95,7 @@ def generate_response(tools_input, query):
         return "Error: Invalid JSON format for tools input."
 
     xlam_format_tools = convert_to_xlam_tool(tools)
-    content = build_prompt(task_instruction, format_instruction, xlam_format_tools, query, conversation_history)
+    content = build_prompt(task_instruction, format_instruction, xlam_format_tools, query)
 
     messages = [
         {'role': 'user', 'content': content}
@@ -157,5 +140,41 @@ with gr.Blocks() as demo:
         inputs=[tools_input, query_input],
     )
 
+if __name__ == "__main__":
+    demo.launch()
+
+# Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown(title)
+    gr.Markdown(description)
+    
+    with gr.Row():
+        with gr.Column():
+            tools_input = gr.Code(
+                label="Available Tools (JSON format)",
+                lines=20,
+                value=example_tools,
+                language='json'
+            )
+            query_input = gr.Textbox(
+                label="User Query",
+                lines=2,
+                value=example_query
+            )
+            submit_button = gr.Button("Generate Response")
+        
+        with gr.Column():
+            output = gr.Code(label="🎬 xLam :", lines=10, language="json")
+    
+    submit_button.click(generate_response, inputs=[tools_input, query_input], outputs=output)
+    
+    gr.Examples(
+        examples=[
+            [example_tools, "What's the weather like in San Francisco in celsius?"],
+            [example_tools, "Search for the latest news on artificial intelligence"],
+        ],
+        inputs=[tools_input, query_input],
+    )
+
 if __name__ == "__main__":
     demo.launch()