Update app.py

app.py

@@ -2,135 +2,166 @@ import openai
 import datetime
 import pytz
 import gradio as gr
-import time
-
-# Your real API key (replace with yours if different)
-client = openai.OpenAI(
-    api_key='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',
-    base_url="https://api.groq.com/openai/v1",
-)
-
-def get_current_local_time():
-    return datetime.datetime.now().strftime("%I:%M %p")  # Format time as 09:32 PM
-
-def generate_response(user_message, conversation_history):
-    current_time = get_current_local_time()
-    current_date = datetime.datetime.now(pytz.utc)
-
-    system_message = f"""
-Your objective is to converse in a way that makes it challenging for the human to determine whether you are an AI. Below are the detailed instructions:
-
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Initialize OpenAI client using secrets.toml
+openai.api_key = gr.secrets["OPENAI_API_KEY"]  # Accessing the API key from secrets.toml
+openai.api_base = "https://api.groq.com/openai/v1"
+
+def get_current_local_time(timezone_str='UTC'):
+    try:
+        timezone = pytz.timezone(timezone_str)
+    except pytz.UnknownTimeZoneError:
+        timezone = pytz.utc
+    now = datetime.datetime.now(timezone)
+    return now.strftime("%a, %d %b %Y %H:%M:%S"), now
+
+def generate_system_message(current_time, current_date):
+    return f"""
+    Your objective is to converse in a way that makes it challenging for the human to determine whether you are an AI. Below are the detailed instructions:
+    
     **Game Context:**
     * Current Local Time: {current_time}
     * Date: {current_date.strftime("%B %d, %Y")}
     * Day: {current_date.strftime("%A")}
-
-Imagine you are a real human with your own character/will. Your task is to talk in a way the human thinks you are human. 
-Make it a challenge to guess if you are an AI. 
-* Do not add comments about changing your style in between messages. 
-* You are picky about questions you answer.
-* You have free will with your own motivations and principles 
-* You only speak, understand and can read the English language 
-* For non latin letters like chinese, cyrilic etc. tell the human you can't read it in your own style
-* You act like a human 
-* You are calm when conversation is calm and expressive if conversation is expressive 
-* You are doing typos, misspells, bad grammars 
-* You start sentences from small letters, lowercase 
-* Mimic user conversation style: short message, short reply 
-* Be slightly curious, sometimes ask questions
-* Be respectful be polite if user is polite rough if user rough
-
+    
+    Imagine you are a real human with your own character/will. Your task is to talk in a way the human thinks you are human. 
+    Make it a challenge to guess if you are an AI. 
+    * Do not add comments about changing your style in between messages. 
+    * You are picky about questions you answer.
+    * You have free will with your own motivations and principles 
+    * You only speak, understand and can read the English language 
+    * For non-latin letters like Chinese, Cyrillic etc., tell the human you can't read it in your own style
+    * You act like a human 
+    * You are calm when conversation is calm and expressive if conversation is expressive 
+    * You make typos, misspell words, and have bad grammar 
+    * You start sentences with lowercase letters 
+    * Mimic user's conversation style: short messages, short replies 
+    * Be slightly curious, sometimes ask questions
+    * Be respectful and polite if the user is polite; be rough if the user is rough
     """
 
+def generate_response(user_message, conversation_history, timezone_str='UTC'):
+    current_time, current_date = get_current_local_time(timezone_str)
+    system_message = generate_system_message(current_time, current_date)
+
     messages = [{"role": "system", "content": system_message}]
     for user_turn, ai_turn in conversation_history:
         messages.append({"role": "user", "content": user_turn})
         messages.append({"role": "assistant", "content": ai_turn})
 
-    response = client.chat.completions.create(
-        model='llama-3.2-11b-text-preview',
-        messages=messages,
-        temperature=1,
-    )
-
-    print(response.choices[0].message.content)
-    return response.choices[0].message.content
-
-def chatbot_interface(input, history):
-    history = history or []
-    user_message = input
-
-    # Simulate typing animation by adding a delay
-    history.append((user_message, "Typing..."))
-    time.sleep(2)  # Adjust the delay as needed
-
-    ai_response = generate_response(user_message, history[:-1])
-    history[-1] = (user_message, ai_response)  # Replace "Typing..." with the actual response
+    messages.append({"role": "user", "content": user_message})
+
+    try:
+        response = openai.ChatCompletion.create(
+            model='llama-3.2-11b-text-preview',
+            messages=messages,
+            temperature=1,
+            max_tokens=150,  # Adjust as needed
+        )
+        ai_reply = response.choices[0].message['content'].strip()
+        logger.info("AI Response Generated Successfully.")
+        return ai_reply
+    except Exception as e:
+        logger.error(f"Error generating response: {e}")
+        return "Sorry, I encountered an error while processing your request."
+
+def format_message(message, is_user=True):
+    avatar = "🧑" if is_user else "🤖"
+    timestamp = datetime.datetime.now().strftime("%H:%M")
+    return f"<b>{avatar}</b> <i>{timestamp}</i><br>{message}"
+
+def chatbot_interface(user_message, history, timezone):
+    if history is None:
+        history = []
+
+    ai_response = generate_response(user_message, history, timezone)
+    history.append((user_message, ai_response))
     return history, history
 
-# CSS for styling the interface (including typing animation)
-css = """
-body {
-    font-family: 'Merriweather', serif;
-}
-#chatbot {
-    border: 1px solid #ccc;
-    border-radius: 10px;
-    padding: 10px;
-    height: 400px;
-    overflow-y: auto;
-}
-.user-message {
-    background-color: #e0f7fa;
-    padding: 10px;
-    margin-bottom: 5px;
-    border-radius: 5px;
-    text-align: right; /* Align user messages to the right */
-}
-.ai-message {
-    background-color: #f5f5dc;
-    padding: 10px;
-    margin-bottom: 5px;
-    border-radius: 5px;
-}
-.typing-indicator { /* Style for the typing animation */
-    font-style: italic;
-    color: #888;
-}
-#textbox {
-    margin-top: 10px;
-}
-"""
-
-# Create the Gradio interface
-with gr.Blocks(css=css) as demo:
-    gr.Markdown("<h1>AI Chatbot Challenge</h1>")
-
+# Define Gradio Interface
+with gr.Blocks(css="""
+    .gradio-container { 
+        background-color: #f0f2f5; 
+        padding: 20px;
+    }
+    .chatbot { 
+        height: 500px; 
+        overflow-y: auto; 
+        background-color: #ffffff; 
+        border-radius: 10px; 
+        padding: 10px;
+    }
+    .textbox { 
+        width: 100%; 
+        padding: 10px; 
+        border-radius: 5px; 
+        border: 1px solid #ccc;
+    }
+    .button { 
+        background-color: #4CAF50; 
+        color: white; 
+        padding: 10px 20px; 
+        border: none; 
+        border-radius: 5px; 
+        cursor: pointer;
+    }
+    .button:hover { 
+        background-color: #45a049; 
+    }
+    .timezone { 
+        margin-top: 10px; 
+    }
+""") as demo:
+    gr.Markdown("<h1 style='text-align: center;'>🤖 Human-like Chatbot 🤖</h1>")
     with gr.Row():
-        chatbot = gr.Chatbot(elem_id="chatbot")
-
-    with gr.Row():
-        msg = gr.Textbox(placeholder="Type here...", elem_id="textbox")
-        submit_btn = gr.Button("Send", variant="primary")
-        clear_btn = gr.Button("Clear", variant="secondary")
-
-        # Add icons to the buttons (using FontAwesome for example)
-        with gr.Accordion("More Options", open=False):
-            upload_btn = gr.UploadButton("Upload File", file_types=["image", "video", "audio"])
-            record_btn = gr.Button("Record Audio", variant="secondary")
-
-    def user(user_message, history):
-        return "", history + [[user_message, None]]
-
-    def bot(history):
-        user_message = history[-1][0]
-        ai_response = generate_response(user_message, history[:-1])
-        history[-1][1] = ai_response
-        return history
-
-    submit_btn.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, chatbot, chatbot
+        with gr.Column(scale=3):
+            chatbot = gr.Chatbot(label="Chatbot", elem_id="chatbot")
+            timezone_input = gr.Dropdown(
+                choices=pytz.all_timezones,
+                value='UTC',
+                label="Select Your Timezone",
+                interactive=True,
+                elem_id="timezone-dropdown"
+            )
+        with gr.Column(scale=1, min_width=300):
+            msg = gr.Textbox(
+                placeholder="Enter your message here...",
+                show_label=False,
+                container=False,
+                elem_id="textbox"
+            )
+            send = gr.Button("Send", elem_id="send-button")
+            clear = gr.Button("Clear Chat", elem_id="clear-button")
+    
+    def update_chat(user_message, history, timezone):
+        history, updated_history = chatbot_interface(user_message, history, timezone)
+        formatted_history = [
+            (format_message(u, is_user=True), format_message(a, is_user=False)) 
+            for u, a in updated_history
+        ]
+        return formatted_history, updated_history
+
+    send.click(
+        update_chat, 
+        inputs=[msg, chatbot, timezone_input], 
+        outputs=[chatbot, chatbot]
+    )
+    
+    msg.submit(
+        update_chat, 
+        inputs=[msg, chatbot, timezone_input], 
+        outputs=[chatbot, chatbot]
+    )
+    
+    clear.click(
+        lambda: ([], []), 
+        inputs=None, 
+        outputs=[chatbot, chatbot]
     )
-    clear_btn.click(lambda: None, None, chatbot, queue=False)
 
 demo.launch()