Update app.py

app.py

@@ -34,7 +34,7 @@ if "follow_up" not in st.session_state:
     st.session_state.follow_up = ""
 
 # ✅ Initialize Hugging Face Model (CPU/GPU Compatible)
-def get_llm_hf_inference(model_id="meta-llama/Llama-2-7b-chat-hf", max_new_tokens=800, temperature=0.6):
+def get_llm_hf_inference(model_id="meta-llama/Llama-2-7b-chat-hf", max_new_tokens=800, temperature=0.3):
     return HuggingFaceEndpoint(
         repo_id=model_id,
         max_new_tokens=max_new_tokens,
@@ -130,7 +130,7 @@ def get_response(system_message, user_text, max_new_tokens=800):
     formatted_chat_history = "\n".join(f"{msg['role']}: {msg['content']}" for msg in chat_history)
 
     # ✅ Invoke Hugging Face Model
-    hf = get_llm_hf_inference(max_new_tokens=max_new_tokens, temperature=0.7)
+    hf = get_llm_hf_inference(max_new_tokens=max_new_tokens, temperature=0.3)
 
     prompt = PromptTemplate.from_template(
         "[INST] You are a helpful AI assistant.\n\nCurrent Conversation:\n{chat_history}\n\n"
@@ -162,6 +162,40 @@ def get_response(system_message, user_text, max_new_tokens=800):
 # ✅ Streamlit UI
 st.title("🚀 HAL - NASA AI Assistant")
 
+# ✅ Justify all chatbot responses
+st.markdown("""
+    <style>
+    .user-msg {
+        background-color: #696969;
+        color: white;
+        padding: 10px;
+        border-radius: 10px;
+        margin-bottom: 5px;
+        width: fit-content;
+        max-width: 80%;
+        text-align: justify;  /* ✅ Justify text */
+    }
+    .assistant-msg {
+        background-color: #333333;
+        color: white;
+        padding: 10px;
+        border-radius: 10px;
+        margin-bottom: 5px;
+        width: fit-content;
+        max-width: 80%;
+        text-align: justify;  /* ✅ Justify text */
+    }
+    .container {
+        display: flex;
+        flex-direction: column;
+        align-items: flex-start;
+    }
+    @media (max-width: 600px) {
+        .user-msg, .assistant-msg { font-size: 16px; max-width: 100%; }
+    }
+    </style>
+""", unsafe_allow_html=True)
+
 # ✅ Display Chat History
 for message in st.session_state.chat_history:
     st.markdown(f"**{message['role'].capitalize()}**: {message['content']}")