fix UI

gradio_anthropic.py

@@ -1,8 +1,7 @@
 import os
+import logging
 
 import gradio as gr
-from openai import OpenAI
-import logging
 import anthropic
 from dotenv import load_dotenv
 

gradio_openai.py

@@ -1,9 +1,11 @@
 import os
+import logging
 
 import gradio as gr
 from openai import OpenAI
-import logging
-import anthropic
+from dotenv import load_dotenv
+
+load_dotenv(".env")
 
 logging.basicConfig(level=logging.INFO)
 logging.getLogger("gradio").setLevel(logging.INFO)
@@ -18,23 +20,19 @@ def generate_completion(input, history):
     messages = [
         {
             "role": "system",
-            "content": "You are a world-class extractor of information from messy job postings.",
+            "content": "You are a world-class assistant.",
         }
     ]
 
-    # Convert history from a list of lists to a list of dictionaries
     if history:
         for entry in history:
-            # Assuming each entry has exactly 2 elements: user input and assistant response
-            if len(entry) == 2:  # Validate format
-                # Append user message
+            if len(entry) == 2:
                 messages.append(
                     {
                         "role": "user",
                         "content": entry[0],
                     }
                 )
-                # Append assistant response
                 messages.append(
                     {
                         "role": "assistant",
@@ -42,7 +40,6 @@ def generate_completion(input, history):
                     }
                 )
 
-    # Append the current user message
     messages.append(
         {
             "role": "user",
@@ -67,7 +64,8 @@ def generate_completion(input, history):
         yield answer_str
 
 
+demo = gr.ChatInterface(fn=generate_completion)
+
 if __name__ == "__main__":
-    demo = gr.ChatInterface(fn=generate_completion)
     demo.queue()
     demo.launch()