Update app.py

app.py

@@ -1,87 +1,40 @@
-import os
-import openai
 import gradio as gr
+import requests
+import os
 
-# Set your OpenAI API key from environment variable
-openai.api_key = os.environ.get("OPENAI_API_KEY")
-
-# System prompt or initial role message
-SYSTEM_PROMPT = {
-    "role": "system", 
-    "content": "You are ZEN AI, a helpful assistant from zenai.world, trained to assist users with valuable insights and solutions related to AI literacy, entrepreneurial strategies, and opportunities for growth."
-}
-
-def process_message(history, user_input, model, temperature):
-    # history: list of tuples (user, assistant)
-    # user_input: current user message
-    # model: model name (e.g., "gpt-4")
-    # temperature: float value for creativity
-
-    # Convert history to the format required by OpenAI
-    messages = [SYSTEM_PROMPT]
-    for user_msg, bot_msg in history:
-        messages.append({"role": "user", "content": user_msg})
-        messages.append({"role": "assistant", "content": bot_msg})
-
-    # Add the new user message
-    messages.append({"role": "user", "content": user_input})
-
-    # Get the response from OpenAI
-    response = openai.ChatCompletion.create(
-        model=model,
-        messages=messages,
-        temperature=temperature,
-        max_tokens=1024
-    )
-
-    # Extract and return the assistant's reply
-    assistant_reply = response.choices[0].message["content"]
-    return assistant_reply
-
-def chat_interface(user_input, history, model, temperature):
-    bot_reply = process_message(history, user_input, model, temperature)
-    history.append((user_input, bot_reply))
-    return history, history
-
-# Define Gradio blocks
-with gr.Blocks(theme="default") as demo:
-    gr.Markdown("<h1 align='center'>ZEN AI Chatbot</h1>")
-    gr.Markdown("Welcome to ZEN AI! Ask me anything about AI literacy, scaling businesses, and connecting with global opportunities.")
-    
-    with gr.Row():
-        model = gr.Dropdown(
-            ["gpt-3.5-turbo", "gpt-4"], 
-            value="gpt-4", 
-            label="OpenAI Model"
-        )
-        temperature = gr.Slider(
-            minimum=0, 
-            maximum=1, 
-            value=0.7, 
-            step=0.1, 
-            label="Temperature"
-        )
+# Retrieve the API key from the environment variables
+API_KEY = os.getenv("OPENAI_API_KEY")
+API_URL = "https://api.openai.com/v1/chat/completions"
 
-    chatbot = gr.Chatbot()
-    state = gr.State([])  # maintain the conversation history
+# Function to call the OpenAI API
+def chat_with_model(prompt):
+    if not API_KEY:
+        return "Error: API key not found. Please set it in the environment variables."
     
-    with gr.Row():
-        user_input = gr.Textbox(
-            lines=1, 
-            placeholder="Ask your question here...",
-            label="Your Message"
-        )
-        submit_button = gr.Button("Submit")
-
-    submit_button.click(
-        chat_interface, 
-        inputs=[user_input, state, model, temperature],
-        outputs=[chatbot, state]
-    )
-
-    # Optionally, allow the user to clear the conversation
-    clear_button = gr.Button("Clear Conversation")
-    clear_button.click(lambda: [], None, chatbot)
-    clear_button.click(lambda: [], None, state)
-
-demo.launch(server_name="0.0.0.0", share=True)
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {API_KEY}"
+    }
+    payload = {
+        "model": "gpt-4",  # Replace with your desired model (e.g., gpt-3.5-turbo, gpt-4)
+        "messages": [{"role": "user", "content": prompt}]
+    }
+
+    try:
+        response = requests.post(API_URL, json=payload, headers=headers)
+        response.raise_for_status()
+        return response.json()["choices"][0]["message"]["content"]
+    except requests.exceptions.RequestException as e:
+        return f"Error: {str(e)}"
+
+# Gradio Interface
+iface = gr.Interface(
+    fn=chat_with_model,
+    inputs="text",
+    outputs="text",
+    title="ZEN AI Chatbot",
+    description="A simple chatbot powered by OpenAI's GPT models."
+)
+
+if __name__ == "__main__":
+    iface.launch()