update

app.py

@@ -2,61 +2,61 @@ import os
 import gradio as gr
 from openai import AzureOpenAI
 
-client = AzureOpenAI(
-    api_key=os.getenv("AZURE_KEY"),
-    api_version=os.getenv("AZURE_VERSION"),
-    azure_endpoint=os.getenv("AZURE_URL")
-)
+# Initialize Azure OpenAI client
+def get_client():
+    return AzureOpenAI(
+        api_key=os.getenv("AZURE_KEY"),
+        api_version=os.getenv("AZURE_VERSION"),
+        azure_endpoint=os.getenv("AZURE_URL")
+    )
+
+# Chat prediction function
+def predict(message, chat_history, company_data, objective, other):
+    client = get_client()
+    chat_history = chat_history or [("", "Hola soy el asistente de Horizon, ¿en qué te puedo ayudar?")]
+
+    # Build system prompt with context
+    system_prompt = (
+        f"You are a helpful assistant. Here is some context about the conversation:\n"
+        f"Company Data: {company_data}\n"
+        f"Objective: {objective}\n"
+        f"Other Information: {other}"
+    )
 
-def chat_with_gpt4o(message, chat_history, company_data, objective, other):
-    chat_history = chat_history or []
-    
-    # If this is the first message, add the initial greeting
-    if not chat_history:
-        initial_greeting = "Hola soy el asistente de Horizon, en que te puedo ayudar?"
-        chat_history.append(("", initial_greeting))
-        return "", chat_history
-    
-    # Create a detailed system prompt with the additional information
-    system_prompt = f"""You are a helpful assistant. Here is some context about the conversation:
-    Company Data: {company_data}
-    Objective: {objective}
-    Other Information: {other}"""
-    
-    # Initialize messages with the enhanced system prompt
+    # Assemble messages
     messages = [{"role": "system", "content": system_prompt}]
-    # Append previous conversation
-    for user, assistant in chat_history:
-        messages.append({"role": "user", "content": user})
-        messages.append({"role": "assistant", "content": assistant})
-    # Add the new user message
+    for user_msg, bot_msg in chat_history:
+        if user_msg:
+            messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
     messages.append({"role": "user", "content": message})
 
-    # Call the Azure OpenAI API
+    # Call Azure OpenAI
     response = client.chat.completions.create(
-        model="gpt-4o",  # Usa el nombre del deployment
+        model="gpt-4o",
         messages=messages
     )
-    assistant_message = response.choices[0].message.content
-
-    # Update chat history
-    chat_history.append((message, assistant_message))
-    # Clear the input box and return updated history
-    return "", chat_history
-
-with gr.Blocks() as demo:
-    with gr.Row():
-        company_data = gr.Textbox(label="Company Data", placeholder="Enter company information...")
-        objective = gr.Textbox(label="Objective", placeholder="Enter the objective...")
-        other = gr.Textbox(label="Other", placeholder="Enter any other relevant information...")
-    
-    with gr.Row():
-        with gr.Column(scale=2):  # Chat section (2/3)
-            chatbot = gr.Chatbot(label="GPT-4o Chat", value=[("", "Hola soy el asistente de Horizon, en que te puedo ayudar?")])
-            msg = gr.Textbox(placeholder="Type your message here...", show_label=False)
-            msg.submit(chat_with_gpt4o, [msg, chatbot, company_data, objective, other], [msg, chatbot])
-        
-        with gr.Column(scale=1):  # Reasoning section (1/3)
-            reasoning = gr.Textbox(label="Reasoning", lines=20, interactive=False)
-    
-    demo.launch()
+    bot_reply = response.choices[0].message.content
+
+    chat_history.append((message, bot_reply))
+    return chat_history, chat_history
+
+# Gradio Interface definition
+app = gr.Interface(
+    fn=predict,
+    inputs=[
+        gr.Textbox(label="Your Message", placeholder="Type your message here..."),
+        gr.State(),
+        gr.Textbox(label="Company Data", placeholder="Enter company information..."),
+        gr.Textbox(label="Objective", placeholder="Enter the objective..."),
+        gr.Textbox(label="Other", placeholder="Enter any other relevant information...")
+    ],
+    outputs=[
+        gr.Chatbot(label="GPT-4o Chat"),
+        gr.State()
+    ],
+    title="Horizon GPT-4o Chat",
+)
+
+if __name__ == "__main__":
+    app.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))