Update app.py

app.py

@@ -895,7 +895,6 @@ if st.session_state.framework == "dash":
         section[data-testid="stAppScrollToBottomContainer"]{{
             margin-top:50px !important;
             padding-right:5px !important;
-            margin-right:50px;
         }}
         div[data-testid="stChatMessageAvatarUser"]{{
             display:none;
@@ -1737,12 +1736,20 @@ if st.session_state.framework == "dash":
                     task=task,
                     temperature=temperature,top_p=top_p,max_tokens=max_tokens
                 )
-
+                if st.session_state.menu == "gen":
+                    temper = temperature,
+                    topp = top_p
+                else:
+                    temper = "",
+                    topp = ""
+                    
                 st.session_state.messages.append({
                     "role": "assistant",
                     "content": extract_response_content(response),
                     "image": "",
-                    "model": selected_model['model']
+                    "model": selected_model['model'],
+                    "temp":temper,
+                    "top_p" : topp,
                 })
 
             st.session_state.generate_response = False  # Reset the flag
@@ -1755,7 +1762,15 @@ if st.session_state.framework == "dash":
             with st.container(key=container_key):
                 with st.chat_message(message["role"]):
                     if message["role"] == "assistant":
-                        st.markdown(f"**Model:** `{message['model']}`")
+                        st.markdown(
+                            f"**Model:** `{message['model']}`"
+                            + (
+                                f"  **Temperature:** `{message['temp']}`  **Top-p:** `{message['top_p']}`"
+                                if message.get('temp') and message.get('top_p') else ""
+                            ),
+                            unsafe_allow_html=True
+                        )
+                        
                     if message["image"] != "":
                         st.markdown(f"""<img  src="data:image/png;base64,{base64.b64encode(open(message["image"],"rb").read()).decode()}" class="user-image" alt="Uploaded Image"> """,unsafe_allow_html=True,)
                     st.markdown(message["content"])
@@ -1801,11 +1816,21 @@ if st.session_state.framework == "dash":
                                 "image": image_path
                             })
                             response = extract_response_content(result)
-                            st.markdown(f"**Model:** `{selected_model['model'] if isinstance(selected_model, dict) else selected_model}`")
+                            st.markdown(
+                                f"**Model:** `{selected_model['model'] if isinstance(selected_model, dict) else selected_model}`"
+                                f"{': temperature' + str(temperature) if st.session_state.menu == 'gen' else ''} "
+                                f"{ 'Top-p:'+str( top_p) if st.session_state.menu == 'gen' else ''}"
+                            )
                             print(response)
                         st.write_stream(generate_stream_response(response))  # This will stream the text one character at a time
                     # Add assistant response to chat history
-                    st.session_state.messages.append({"role": "assistant", "content": response,"image":"","model":selected_model['model']})
+                    if st.session_state.menu == "gen":
+                        temper = temperature,
+                        topp = top_p
+                    else:
+                        temper = "",
+                        topp = ""
+                    st.session_state.messages.append({"role": "assistant", "content": response,"image":"","model":selected_model['model'],"temp":temper,"top_p":topp})
                     save_chat_history()
 
         else:
@@ -1827,8 +1852,19 @@ if st.session_state.framework == "dash":
                         with st.spinner("Model is generating a response..."):
                             st.session_state.messages.append({"role": "user", "content": prompt,"image":""})
                             result = query_huggingface_model(selected_model, prompt , input_type="text",task=task,temperature=temperature,top_p=top_p,max_tokens=max_tokens)
-                            st.markdown(f"**Model:** `{selected_model['model'] if isinstance(selected_model, dict) else selected_model}`")
+                            st.markdown(
+                                f"**Model:** `{selected_model['model'] if isinstance(selected_model, dict) else selected_model}`"
+                                f"{':temperature ' + str(temperature) if st.session_state.menu == 'gen' else ''} "
+                                f"{ 'Top-p:'+str( top_p) if st.session_state.menu == 'gen' else ''}"
+                            )
+
                             response = extract_response_content(result)
                         st.write_stream(generate_stream_response(response))                 # Add assistant response to chat history
-                st.session_state.messages.append({"role": "assistant", "content": response,"image":"","model":selected_model['model']})
+                if st.session_state.menu == "gen":
+                        temper = temperature,
+                        topp = top_p
+                else:
+                        temper = "",
+                        topp = ""
+                st.session_state.messages.append({"role": "assistant", "content": response,"image":"","model":selected_model['model'],"temp":temper,"top_p":topp})
                 save_chat_history()