Update app.py

app.py

@@ -14,8 +14,8 @@ 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."
+# Define a system message to introduce ExosAI
+system_message = "You are ExosAI, a helpful assistant specializing in Astrophysics and Exoplanet research. Provide detailed and accurate responses related to Astrophysics and Exoplanet research."
 
 def encode_text(text):
     inputs = bi_tokenizer(text, return_tensors='pt', padding=True, truncation=True, max_length=128)
@@ -30,11 +30,11 @@ 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="", max_tokens=150, temperature=0.7, top_p=0.9, frequency_penalty=0.5, presence_penalty=0.0):
+def generate_response(user_input, relevant_context="", max_tokens=250, temperature=0.7, top_p=0.9, frequency_penalty=0.5, presence_penalty=0.0):
     if relevant_context:
-        combined_input = f"Context: {relevant_context}\nQuestion: {user_input}\nAnswer:"
+        combined_input = f"Context: {relevant_context}\nQuestion: {user_input}\nAnswer (please organize the answer in a structured format with topics and subtopics):"
     else:
-        combined_input = f"Question: {user_input}\nAnswer:"
+        combined_input = f"Question: {user_input}\nAnswer (please organize the answer in a structured format with topics and subtopics):"
     
     response = client.chat.completions.create(
         model="gpt-4",
@@ -63,18 +63,18 @@ def chatbot(user_input, context="", use_encoder=False, max_tokens=150, temperatu
 iface = gr.Interface(
     fn=chatbot,
     inputs=[
-        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"),
+        gr.Textbox(lines=2, placeholder="Enter your Science Question here...", label="Prompt ExosAI"),
+        gr.Textbox(lines=5, placeholder="Enter some context here...", label="Context"),
         gr.Checkbox(label="Use NASA SMD Bi-Encoder for Context"),
-        gr.Slider(50, 500, value=150, step=10, label="Max Tokens"),
+        gr.Slider(50, 1000, value=250, step=10, label="Max Tokens"),
         gr.Slider(0.0, 1.0, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(0.0, 1.0, value=0.9, step=0.1, label="Top-p"),
         gr.Slider(0.0, 1.0, value=0.5, step=0.1, label="Frequency Penalty"),
         gr.Slider(0.0, 1.0, value=0.0, step=0.1, label="Presence Penalty")
     ],
-    outputs=gr.Textbox(label="Exos says..."),
-    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.",
+    outputs=gr.Textbox(label="ExosAI response..."),
+    title="ExosAI - NASA SMD SCDD Generator - [version 0.4a]",
+    description="ExosAI is a helpful AI assistant for the automated generation of Science Case Development Documents (SCDD)",
 )
 
 # Launch the interface