Fix formating

app.py

@@ -225,14 +225,14 @@ def generate(input_message: str, db_info="", temperature=0.2, top_p=0.9, top_k=0
             # Test against dummy database
             db_code = generate_dummy_db(db_info, input_message, query)
             success = test_query_on_dummy_db(db_code, query)
+            # Format again
+            query = format(query) if format_sql else query
             if success:
                 responses.append(query)
         else:
             responses.append(query)
             
-
-    # Concat responses to be a single string seperated by a newline
-    #output = "\n".join(responses)
+    # Choose the first response
     output = responses[0] if responses else ""
 
     if log: