Update app.py

app.py

@@ -1,6 +1,3 @@
-# Install necessary libraries (if running locally)
-# !pip install transformers torch textblob numpy gradio
-
 # Import necessary libraries
 import torch
 import torch.nn as nn
@@ -108,7 +105,7 @@ cnn_model = CNN(vocab_size=len(tokenizer), embedding_dim=100, num_filters=64, fi
 ffnn_model = FFNN(input_dim=100, hidden_dim=50, output_dim=1).to(device)
 
 # ---- Response Generation ----
-def generate_response(prompt, max_length=1024):
+def generate_response(prompt, max_length=512):
     inputs = tokenizer(prompt, return_tensors='pt', padding=True, truncation=True, max_length=max_length)
     input_ids = inputs['input_ids'].to(device)
     attention_mask = inputs['attention_mask'].to(device)
@@ -130,7 +127,25 @@ def generate_response(prompt, max_length=1024):
         )
 
     response = tokenizer.decode(output[0], skip_special_tokens=True)
-    return response.strip()
+
+    # Look for the first indentation or newline (i.e., the first significant break in the response)
+    parts = response.split("\n", 1)
+
+    if len(parts) > 1:
+        # Text before the first indentation is orange
+        before_indent = f'<span style="color: orange;">{parts[0].strip()}</span>'
+
+        # Text after the first newline or indentation gets labeled as "Inner Thoughts:" in blue
+        after_indent = f'<span style="color: blue;">Inner Thoughts: {parts[1].strip()}</span>'
+        
+        # Combine the two parts
+        colored_response = before_indent + '\n' + after_indent
+    else:
+        # If there's no newline, color the entire response orange
+        colored_response = f'<span style="color: orange;">{response.strip()}</span>'
+
+    return colored_response
+
 
 # ---- Interactive Chat Function ----
 def advanced_agi_chat(user_input):
@@ -153,6 +168,7 @@ def advanced_agi_chat(user_input):
 # ---- Gradio Interface ----
 def chat_interface(user_input):
     response = advanced_agi_chat(user_input)
+    # Return the response with HTML formatting
     return response
 
 with gr.Blocks() as app:
@@ -162,7 +178,7 @@ with gr.Blocks() as app:
             user_input = gr.Textbox(label="What will you say to Gertrude?", placeholder="Type something here... Expect 1-2 Minute Response Times...")
             submit_button = gr.Button("Send")
         with gr.Column():
-            chatbot = gr.Textbox(label="Gertrude's Response", interactive=False)
+            chatbot = gr.HTML(label="Gertrude's Response", interactive=False)
     
     # Theme the UI with colors and a pattern
     gr.HTML("<style>.gradio-container { background-color: #F4F8FF; padding: 20px; border-radius: 15px; font-family: 'Comic Sans MS'; }</style>")