prompt edits

app.py

@@ -46,7 +46,7 @@ def generate_response(user_message): #generate_response_token_by_token
         if token_buffer:
             elapsed_time = time.time() - start_time  # Calculate elapsed time
             alltokens += token_buffer
-            yield f"{alltokens} \n\n [Inference time: {elapsed_time:.2f} seconds | Average Tokens per second: { tokencount / elapsed_time}]"
+            yield f"{alltokens} \n\n [Inference time: {elapsed_time:.2f} seconds | Average Tokens per second: { tokencount / elapsed_time }]"
 
     finally:
         try:
@@ -67,15 +67,15 @@ def generate_response(user_message): #generate_response_token_by_token
 
 
 def custom_generate_response(cust_user_message):
-    cust_user_message = CustomPrompts[0] + '\n\n' + cust_user_message + '\n\n'
+    cust_user_message = CustomPrompts[0] + '\n\n' + cust_user_message + '\n\nClass Diagram:'
     yield from generate_response(cust_user_message)
 
 def custom_generate_response1(cust_user_message):
-    cust_user_message = CustomPrompts[1] + '\n\n' + cust_user_message + '\n\n'
+    cust_user_message = CustomPrompts[1] + '\n\n' + cust_user_message + '\n\nPydot Code:'
     yield from generate_response(cust_user_message)
 
 def custom_generate_response2(cust_user_message):
-    cust_user_message = CustomPrompts[2] + '\n' + cust_user_message + '\n\n'
+    cust_user_message = CustomPrompts[2] + '\n' + cust_user_message + '\n\nScene Details'
     yield from generate_response(cust_user_message)
 
 CustomPrompts = [