Update app.py

app.py

@@ -45,7 +45,12 @@ if not HF_TOKEN:
 
 # --- UI Settings ---
 TITLE = "<h1 style='text-align:center; margin-bottom: 20px;'>Local RAG Llama 3.1 8B</h1>"
-<a href="https://discord.gg/openfreeai" target="_blank"><img src="https://img.shields.io/static/v1?label=Discord&message=Openfree%20AI&color=%230000ff&labelColor=%23800080&logo=discord&logoColor=white&style=for-the-badge" alt="badge"></a>
+DISCORD_BADGE = """<p style="text-align:center; margin-top: -10px;">
+<a href="https://discord.gg/openfreeai" target="_blank">
+  <img src="https://img.shields.io/static/v1?label=Discord&message=Openfree%20AI&color=%230000ff&labelColor=%23800080&logo=discord&logoColor=white&style=for-the-badge" alt="badge">
+</a>
+</p>
+"""
 
 CSS = """
 .upload-section {
@@ -222,21 +227,14 @@ def create_or_update_index(files, request: gr.Request):
         }
     
     node_parser = HierarchicalNodeParser.from_defaults(
-        chunk_sizes=[2048, 512, 128],
-        chunk_overlap=20
+        chunk_sizes=[2048, 512, 128],  
+        chunk_overlap=20         
     )
     logger.info(f"Parsing {len(new_documents)} documents into hierarchical nodes")
     new_nodes = node_parser.get_nodes_from_documents(new_documents)
     new_leaf_nodes = get_leaf_nodes(new_nodes)
     new_root_nodes = get_root_nodes(new_nodes)
     logger.info(f"Generated {len(new_nodes)} total nodes ({len(new_root_nodes)} root, {len(new_leaf_nodes)} leaf)")
-    node_ancestry = {}
-    for node in new_nodes:
-        if hasattr(node, 'metadata') and 'file_name' in node.metadata:
-            file_origin = node.metadata['file_name']
-            if file_origin not in node_ancestry:
-                node_ancestry[file_origin] = 0
-            node_ancestry[file_origin] += 1
     
     if os.path.exists(save_dir):
         logger.info(f"Loading existing index from {save_dir}")
@@ -411,9 +409,11 @@ def stream_chat(
 
 def create_demo():
     with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
-        # Title only
+        # Title
         gr.HTML(TITLE)
-        
+        # Discord badge immediately under the title
+        gr.HTML(DISCORD_BADGE)
+
         with gr.Row(elem_classes="main-container"):
             with gr.Column(elem_classes="upload-section"):
                 file_upload = gr.File(
@@ -457,7 +457,7 @@ def create_demo():
                 
                 with gr.Accordion("Advanced Settings", open=False):
                     system_prompt = gr.Textbox(
-                        value="You are a deep thinking AI, you may use extremely long chains of thought to deeply consider the problem and deliberate with yourself via systematic reasoning processes to help come to a correct solution prior to answering. You should enclose your thoughts and internal monologue inside tags, and then provide your solution or response to the problem. As a knowledgeable assistant, provide detailed answers using the relevant information from all uploaded documents.",
+                        value="You are a deep thinking AI, you may use extremely long chains of thought to deeply consider the problem and deliberate with yourself via systematic reasoning processes to help come to a correct solution prior to answering. You should enclose your thoughts and internal monologue inside tags, and then provide your solution or response to the problem. As a knowledgeable assistant, provide detailed answers using the relevant information from all uploaded documents.",                        
                         label="System Prompt",
                         lines=3
                     )
@@ -474,7 +474,7 @@ def create_demo():
                             minimum=128,
                             maximum=8192,
                             step=64,
-                            value=4000,
+                            value=1024,
                             label="Max New Tokens",
                         )
                         top_p = gr.Slider(