Update app.py

app.py

@@ -51,7 +51,6 @@ if  st.session_state.show_overlay == False:
 else:
     top= -7
     left =-5
-
 # Folder to store chat histories
 CHAT_DIR = "chat_histories"
 os.makedirs(CHAT_DIR, exist_ok=True)
@@ -67,7 +66,7 @@ def save_chat_history():
 
 # Function to load data
 
-def query_huggingface_model(selected_model: dict, input_data, input_type="text",max_new_tokens=512,task="text-classification"):
+def query_huggingface_model(selected_model: dict, input_data, input_type="text",max_tokens=512,task="text-classification",temperature=0.7, top_p=0.9 ):
     API_URL = selected_model.get("url")
     headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
@@ -81,7 +80,9 @@ def query_huggingface_model(selected_model: dict, input_data, input_type="text",
                             "content": input_data
                         }
                     ],                    
-                    "max_tokens": max_new_tokens,
+                    "max_tokens": max_tokens,
+                    "temperature": temperature,
+                    "top_p": top_p,
                     "model":selected_model.get("model")
                 }
                     
@@ -617,6 +618,7 @@ if st.session_state.framework == "dash":
         image3 = 'images.png'
         margin = 0
         margintop = "-20"
+        display = "none"
         st.session_state.models = [
     {
         "model": "distilbert-base-uncased-finetuned-sst-2-english",
@@ -654,6 +656,8 @@ if st.session_state.framework == "dash":
         margin = 0
         margintop = -20
         task = "text-generation"
+        display = "block"
+
         st.session_state.models = [
     {
         "model": "deepseek-ai/DeepSeek-V3",
@@ -688,6 +692,8 @@ if st.session_state.framework == "dash":
         image2 = 'hugging.png'
         margintop = -90
         image3 = 'Captured.png'
+        display = "none"
+
         st.session_state.models = [
     {
         "model": "Salesforce/blip-image-captioning-large",
@@ -852,11 +858,11 @@ if st.session_state.framework == "dash":
             /*border-bottom:none !important;*/
            margin-right: 300px !important;
             margin-top: 12.5px !important;
-            z-index: 11 !important; /* Ensure it stays above other elements */
             border-bottom:1px solid #ccc;
-            width:64%;
+            width:48%;
             height:85px;
-            margin-left:520px;
+            margin-left:770px;
+            z-index:1000000;
             
 
         }}
@@ -868,7 +874,7 @@ if st.session_state.framework == "dash":
         border:none;
         }}
         div[data-testid="stApp"]{{
-         background:#f5f3f0;
+         background:white;
         height: 98.9vh;  /* Full viewport height */
         width: 99%; 
         border-radius: 10px !important;
@@ -887,8 +893,8 @@ if st.session_state.framework == "dash":
             
         }}
         section[data-testid="stAppScrollToBottomContainer"]{{
-            margin-top:80px !important;
-            padding-right:35px !important;
+            margin-top:50px !important;
+            padding-right:5px !important;
         }}
         div[data-testid="stChatMessageAvatarUser"]{{
             display:none;
@@ -950,7 +956,7 @@ if st.session_state.framework == "dash":
 
         }}
     [class*="st-key-content_1"] {{
-            background: white;
+            background: transparent;
             border-radius: 10px;
           /*  box-shadow: 0px 4px 10px rgba(0, 0, 0, 0.1);*/
             width: 90%;
@@ -964,11 +970,11 @@ if st.session_state.framework == "dash":
             position: fixed; /* Fix the position of the container */
             top: 1%; /* Adjust top offset */
             left: 15%; /* Adjust left offset */
-            height: 96.5vh;  /* Full viewport height */
             overflow-x:hidden;
             overflow-y:hidden;
              display: flex;
              
+            z-index:10000;
 
         }}
         
@@ -976,7 +982,7 @@ if st.session_state.framework == "dash":
         div[data-testid="stSidebarHeader"]{{
             height:40px !important;
             position:fixed;
-            z-index:100000000000;
+            z-index:10000;
             background:#f9f9f9;
             width:300px;
         }}
@@ -1246,9 +1252,33 @@ if st.session_state.framework == "dash":
                 padding:15px;
                 padding-left:100px;
 
+                
+
+             }}
+            div[data-testid="stExpander"]
+        {{
+                width: 250px !important;
+                background-color: white !important; 
+                margin-top: -102px !important;
+                position: relative;
+                margin-left: 289px !important;
+                z-index:10000000000000000000;
+                position:fixed;
+                border-bottom:1px solid #ccc;
+                border-radius:0px;
+                height:87px;
+                padding-top:35px;
+                padding-left:10px;
+
 =                
 
              }}
+             div[data-testid="stExpander"]>details{{
+               z-index:10000000000000000000;
+
+              display:{display} !important;
+        background:white;
+        }}
          div[data-testid="stTextInput"]
         {{
                 background-color: transparent;  /* White background */
@@ -1280,7 +1310,7 @@ if st.session_state.framework == "dash":
             width:40px !important;
             margin-top:{top}% !important;
             margin-left:{left}% !important;
-            z-index:10000000;
+            z-index:1000000000000;
             background-image: url(data:image/png;base64,{base64.b64encode(open("side.png", "rb").read()).decode()});
             background-size: contain; /* size of the image */
             background-repeat: no-repeat;
@@ -1518,6 +1548,10 @@ if st.session_state.framework == "dash":
         model_names = [m["model"] for m in st.session_state.models]
         selected_model_name = st.selectbox(task, model_names)
         selected_model = next((m for m in st.session_state.models if m["model"] == selected_model_name), None)
+        with st.expander("⚙️ Advanced Settings"):
+                temperature = st.slider("Temperature", 0.0, 2.0, 0.7, step=0.1)
+                top_p = st.slider("Top-p ", 0.0, 1.0, 0.9, step=0.05)
+                max_tokens = st.slider("Max tokens", 50, 1024, 512, step=50)
         if st.session_state.show_overlay == True:
             st.header(f"{text_h1} – What can I help you with?")
             user_input = ''
@@ -1699,7 +1733,8 @@ if st.session_state.framework == "dash":
                     selected_model,
                     st.session_state.input_text,                
                     input_type=st.session_state.input_task,
-                    task=task
+                    task=task,
+                    temperature=temperature,top_p=top_p,max_tokens=max_tokens
                 )
 
                 st.session_state.messages.append({
@@ -1790,7 +1825,7 @@ if st.session_state.framework == "dash":
                     with st.chat_message("assistant"):
                         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)
+                            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}`")
                             response = extract_response_content(result)
                         st.write_stream(generate_stream_response(response))                 # Add assistant response to chat history