Update app.py

app.py

@@ -14,6 +14,9 @@ bi_model = AutoModel.from_pretrained(bi_encoder_model_name)
 api_key = os.getenv('OPENAI_API_KEY')
 client = OpenAI(api_key=api_key)
 
+# Define a system message to introduce Exos
+system_message = "You are Exos, a helpful assistant specializing in Exoplanet research. Provide detailed and accurate responses related to Exoplanet research."
+
 def encode_text(text):
     inputs = bi_tokenizer(text, return_tensors='pt', padding=True, truncation=True, max_length=128)
     outputs = bi_model(**inputs)
@@ -27,12 +30,16 @@ def retrieve_relevant_context(user_input, context_texts):
     most_relevant_idx = np.argmax(similarities)
     return context_texts[most_relevant_idx]
 
-def generate_response(user_input, relevant_context):
-    combined_input = f"Context: {relevant_context}\nQuestion: {user_input}\nAnswer:"
+def generate_response(user_input, relevant_context=""):
+    if relevant_context:
+        combined_input = f"Context: {relevant_context}\nQuestion: {user_input}\nAnswer:"
+    else:
+        combined_input = f"Question: {user_input}\nAnswer:"
     
     response = client.chat.completions.create(
         model="gpt-4",
         messages=[
+            {"role": "system", "content": system_message},
             {"role": "user", "content": combined_input}
         ],
         max_tokens=150,
@@ -41,11 +48,14 @@ def generate_response(user_input, relevant_context):
         frequency_penalty=0.5,
         presence_penalty=0.0
     )
-    return response.choices[0].message.content.strip()
-
-def chatbot(user_input, context=""):
-    context_texts = context.split("\n")
-    relevant_context = retrieve_relevant_context(user_input, context_texts) if context else ""
+    return response.choices[0].message['content'].strip()
+
+def chatbot(user_input, context="", use_encoder=False):
+    if use_encoder and context:
+        context_texts = context.split("\n")
+        relevant_context = retrieve_relevant_context(user_input, context_texts)
+    else:
+        relevant_context = ""
     response = generate_response(user_input, relevant_context)
     return response
 
@@ -53,12 +63,13 @@ def chatbot(user_input, context=""):
 iface = gr.Interface(
     fn=chatbot,
     inputs=[
-        gr.Textbox(lines=2, placeholder="Enter your message here..."),
-        gr.Textbox(lines=5, placeholder="Enter context here, separated by new lines...")
+        gr.Textbox(lines=2, placeholder="Enter your message here...", label="Your Question"),
+        gr.Textbox(lines=5, placeholder="Enter context here, separated by new lines...", label="Context (Optional)"),
+        gr.Checkbox(label="Use Bi-Encoder for Context")
     ],
     outputs="text",
-    title="Context-Aware Dynamic Response Chatbot",
-    description="A chatbot using a NASA-specific bi-encoder model to understand the input context and GPT-4 to generate dynamic responses. Enter context to get more refined and relevant responses."
+    title="Exos - Your Exoplanet Research Assistant",
+    description="Exos is a helpful assistant specializing in Exoplanet research. Provide context to get more refined and relevant responses.",
 )
 
 # Launch the interface
@@ -73,3 +84,4 @@ iface.launch(share=True)
 
 
 
+