llm

app.py

@@ -1,21 +1,56 @@
+import os
 import gradio as gr
+from openai import OpenAI
+
+client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
+
+FORMAT_INSTRUCTION = """
+Please format your response exactly as:
+
+Intent:
+Priority:
+Owner:
+Action Required:
+Reasoning:
+Draft Reply:
+"""
 
 def run_agent(system_prompt, email):
-    return f"System Prompt:\n{system_prompt}\n\nEmail:\n{email}"
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": email + "\n\n" + FORMAT_INSTRUCTION},
+            ],
+            temperature=0.4,
+        )
+        return response.choices[0].message.content
+
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             gr.Markdown("## Agent Builder")
-            system_prompt = gr.Textbox(label="System Prompt", lines=15)
+            system_prompt = gr.Textbox(
+                label="System Prompt",
+                lines=15,
+                value="You are an assistant that processes emails."
+            )
 
         with gr.Column():
             gr.Markdown("## Email Input")
-            email = gr.Textbox(label="Email", lines=15)
+            email = gr.Textbox(
+                label="Email",
+                lines=15,
+                value="We have not received our shipment and this is causing major delays. Please respond urgently."
+            )
 
         with gr.Column():
             gr.Markdown("## Output")
-            output = gr.Textbox(label="Result", lines=15)
+            output = gr.Markdown()
 
     run_btn = gr.Button("Run Agent")
     run_btn.click(run_agent, inputs=[system_prompt, email], outputs=output)

requirements.txt

@@ -1 +1,2 @@
-gradio
+gradio
+openai