Update app.py

app.py

@@ -1,11 +1,24 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
 import datetime
+import re
+import requests
+import json
 
 # Initialize the InferenceClient
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-def format_prompt(message, history):
+# Enter your API key here
+api_key = "1e8cd0385845a649e448cde4917058d6"
+
+# Define the system prompt templates
+system_prompt_templates = {
+    r"\btime\b|\bhour\b|\bclock\b": "server log: ~This message was sent at {formatted_time}. The actual year is 2024.~",
+    r"\bweather\b|\bforecast\b|\bmeteo": "server log: ~The current weather conditions in {city_name} are {weather_description} with a high of {current_temperature_c}°C ({current_temperature_f}°F) and a pressure of {current_pressure_hpa} hPa ({current_pressure_inHg} inHg) and humidity of {current_humidity}%.~",
+    r"\bdate\b|\bcalendar\b": "server log: ~Today's date is {formatted_date}.~",
+}
+
+def format_prompt(message, history, system_prompt):
     prompt = "<s>"
     for user_prompt, bot_response in history:
         prompt += f"\[INST\] {user_prompt} \[/INST\]"
@@ -25,10 +38,47 @@ def generate(prompt, history, system_prompt, temperature=0.9, max_new_tokens=904
         seed=42,
     )
 
-    # Get current time
+    # Get current time and date
     now = datetime.datetime.now()
     formatted_time = now.strftime("%H.%M.%S, %B, %Y")
-    system_prompt = f"server log: ~This message was sent at {formatted_time}. The actual year is 2024.~"
+    formatted_date = now.strftime("%B %d, %Y")
+
+    # Check for keywords in the user's input and update the system prompt accordingly
+    city_name = None
+    weather_description = None
+    current_temperature_c = None
+    current_temperature_f = None
+    current_pressure_hpa = None
+    current_pressure_inHg = None
+    current_humidity = None
+    for keyword, template in system_prompt_templates.items():
+        if re.search(keyword, prompt, re.IGNORECASE):
+            if keyword == r"\bweather\b|\bforecast\b|\bmeteo":
+                base_url = "http://api.openweathermap.org/data/2.5/weather?"
+                complete_url = base_url + "appid=" + api_key + "&q=" + city_name
+                response = requests.get(complete_url)
+                x = response.json()
+                if x["cod"] != "404":
+                    y = x["main"]
+                    current_temperature_c = y["temp"] - 273.15  # Convert from Kelvin to Celsius
+                    current_temperature_f = current_temperature_c * 9/5 + 32  # Convert from Celsius to Fahrenheit
+                    current_pressure_hpa = y["pressure"]
+                    current_pressure_inHg = current_pressure_hpa * 0.02953  # Convert from hPa to inHg
+                    current_humidity = y["humidity"]
+                    z = x["weather"]
+                    weather_description = z[0]["description"]
+                    city_name = x["name"]
+                else:
+                    print("City Not Found")
+                    city_name = "unknown"
+                    weather_description = "unknown"
+                    current_temperature_c = 0
+                    current_temperature_f = 32
+                    current_pressure_hpa = 0
+                    current_pressure_inHg = 0
+                    current_humidity = 0
+            system_prompt = template.format(formatted_time=formatted_time, formatted_date=formatted_date, city_name=city_name, weather_description=weather_description, current_temperature_c=current_temperature_c, current_temperature_f=current_temperature_f, current_pressure_hpa=current_pressure_hpa, current_pressure_inHg=current_pressure_inHg, current_humidity=current_humidity)
+            break
 
     formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
@@ -45,10 +95,29 @@ additional_inputs = [
     gr.Slider(label="Repetition penalty", value=1.2, minimum=1.0, maximum=2.0, step=0.05, interactive=True, info="Penalize repeated tokens")
 ]
 
-gr.ChatInterface(
-    fn=generate,
-    chatbot=gr.Chatbot(show_label=True, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
-    additional_inputs=additional_inputs,
-    title="ConvoLite",
-    concurrency_limit=20,
-).launch(show_api=False)
+def check_keywords(text):
+    for keyword, _ in system_prompt_templates.items():
+        if re.search(keyword, text, re.IGNORECASE):
+            return True
+    return False
+
+chatbot = gr.Chatbot(show_label=True, show_share_button=False, show_copy_button=True, likeable=True, layout="panel")
+with gr.Blocks():
+    with gr.Row():
+        with gr.Column(scale=3):
+            user_input = gr.Textbox(label="Your message", placeholder="Type your message here...")
+        with gr.Column(scale=1):
+            submit_button = gr.Button("Send")
+
+    with gr.Row():
+        chatbot_output = chatbot
+
+    submit_button.click(
+        fn=generate,
+        inputs=[user_input, chatbot, gr.Textbox(label="System Prompt", max_lines=1, interactive=True)],
+        outputs=chatbot_output,
+        every=200,
+        _js="check_keywords"
+    )
+
+gr.Blocks().launch(show_api=False)